corner
emcee
GOFevaluation
graphviz
h5py
immutabledict
jax
matplotlib
multihist
numpy<2.0.0
numpyro
pandas
scikit-learn
scipy
strax
straxen

[cpu]
jax[cpu]

[cuda112]
jax[cuda]==0.3.15

[cuda121]
jax[cuda12_pip]
