NUTS: Getting the Discarded Samples


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


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).