jax
numpyro
pandas
seaborn
tensorflow_probability
arviz

[cpu]
jax[cpu]

[cuda12]
jax[cuda12]
