Getting only the warmup samples

numpyro
2

See NUTS: Getting the Discarded Samples

1 Like