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

[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

[gpu]
jax[cuda13]

[grb]
afterglowpy
