numpyro==0.18.0
numpy==1.26.3
matplotlib==3.8.2
pandas==2.2.3
seaborn==0.13.2
tensorflow_probability==0.24.0
arviz==0.17.0
funsor==0.4.5
IPython==8.12.3

[cpu]
jax==0.5.1
jaxlib==0.5.1

[gpu]
jax[cuda12_pip]==0.5.1
