jax
numpyro
pandas
seaborn
tensorflow_probability
arviz

[cuda]
jax[cuda12]
