numpy>=1.24.0
jax>=0.4.20
sncosmo>=2.9.0
astropy>=5.0.0
pyyaml>=6.0.0
matplotlib
tqdm
requests

[cuda12]
jax[cuda12]>=0.4.20

[dev]
pytest>=7.0
pytest-cov>=2.0
black>=22.0
isort>=5.0

[docs]
sphinx>=5.0.0
sphinx-rtd-theme>=1.0.0
sphinxcontrib-mermaid>=0.7.0
sphinx-autodoc-typehints>=1.12.0

[nested]
anesthetic
blackjax>=1.0.0
distrax
