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

[cpu]
jax[cpu]

[cuda112]
jax[cuda]==0.3.15

[cuda121]
jax[cuda12_pip]
