Max_steps with sampling_jax.sample_numpyro_nuts

It is not guarantee. You situation is most likely some parameters is/are stuck at difficult geometry, you should look at modifying the prior and inspect the posterior before increasing the number of steps.

1 Like