Ah, you’re using jax. Does constant data appear if you use pm.sample()? If so, it may be similar to (but different from) this issue.
Ah, you’re using jax. Does constant data appear if you use pm.sample()? If so, it may be similar to (but different from) this issue.