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?
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?