dill
jax>=0.4.14
jaxlib>=0.4.14
flax
numpy
pandas
jaxtyping
beartype
tqdm
scipy
ml_collections
matplotlib
astropy
sncosmo
h5py
corner
huggingface_hub>=0.13.0

[blackjax]
blackjax

[docs]
sphinx>=7.0
sphinx-autobuild
sphinx-book-theme>=1.0
sphinx-design>=0.5
sphinx-autodoc-typehints>=2.0
sphinx-copybutton>=0.5
myst-parser>=2.0
dust-extinction

[flowmc]
flowMC>=0.6.0

[gpu]
jax[cuda13]

[grb]
afterglowpy

[numpyro]
numpyro

[samplers]
fiestaEM[blackjax,flowmc,numpyro]
