corner>=2.2.3
diffrax>=0.7.0
einops>=0.8.1
equinox>=0.13.1
flax>=0.12.0
grain>=0.2.12
jax>=0.7.2
matplotlib>=3.10
numpy>=2.3
numpyro>=0.19.0
optax>=0.2.6
orbax-checkpoint>=0.11.25
seaborn>=0.13.2
tqdm>=4.62.0
