NUTS: Getting the Discarded Samples

Hello

Is it possible to also save the (warm-up) samples which should be discard in the resulting dictionary with posterior samples?

Sunny

Yes, you can run warmup method first with collect_warmup=True, get those warmup samples, and then run the sampling phase (this will skip the warmup phase).