Error on running out of sample sampling

Hm, that sounds like a bug, but I’d be surprised if that kind of bug slipped through our test pipeline.

Can you reduce the example to something minimal?
What if you don’t use the sampling_jax but regular sampling?