numpyro
numpy
matplotlib
pandas
seaborn
tensorflow_probability
arviz
funsor
plotly
IPython

[cpu]
jax
jaxlib

[gpu]
jax[cuda12_pip]
