Random sampling JAX failures caused by `jax>=0.4.5`
brandonwillard opened this issue · comments
It looks like a new version of JAX has resulted in errors in at least one sampling test and it's blocking CI runs.
aesara-devs / aesara
Aesara is a Python library for defining, optimizing, and efficiently evaluating mathematical expressions involving multi-dimensional arrays.