jax>=0.4.30
jaxlib>=0.4.30
numpy>=1.26

[gpu]
jax[cuda12]>=0.4.30

[notebook]
matplotlib>=3.8
ipykernel>=6.29
numpyro>=0.15

[numpyro]
numpyro>=0.15
