numpy
scipy
astropy
healpy
pyyaml
jax>=0.9
optax>=0.2.5
ggah_mod>=1.0

[cosmo]
blackjax>=1.2

[cuda]
jax[cuda12]>=0.9

[cuda13]
jax[cuda13]>=0.9

[dev]
pytest
pytest-cov
coverage>=7.10
matplotlib

[docs]
sphinx>=7
sphinx-rtd-theme
myst-nb>=1.0
sphinxcontrib-bibtex>=2.6

[plots]
matplotlib
