numpyro
numpy
matplotlib
pandas
seaborn
tensorflow_probability
arviz
funsor
scip
plotly

[cpu]
jax==0.6.2
jaxlib==0.6.2

[gpu]
jax[cuda12_pip]==0.6.2
