jaxlib>=0.9.0
numpy>=2.0
scipy>=1.13
matplotlib>=3.9
h5py>=3.16
astropy>=8.0
tensorflow-probability>=0.25
blackjax>=1.6
tqdm>=4.65
optax>=0.2.3
anesthetic>=2.14
pytest>=7.0

[:sys_platform != "linux"]
jax>=0.9.0

[:sys_platform == "linux"]
jax[cuda12]>=0.9.0

[docs]
mkdocs-material>=9.5
mkdocstrings[python]>=0.24
black>=24.0
