numpyro
pandas
seaborn
tensorflow_probability
arviz
funsor
IPython

[cpu]
jax[cpu]

[gpu]
jax[cuda12_pip]
