numpyro==0.18.0
pandas
seaborn
tensorflow_probability==0.25.0
arviz
funsor
IPython

[cpu]
jax==0.5.1
jaxlib==0.5.1

[gpu]
jax[cuda12_pip]==0.5.1
