jax
jaxlib
zodiax>=0.4.1

[dev]
pytest
black
ruff
pre-commit
mkdocs
mkdocs-jupyter
mkdocs-same-dir
mkdocs-autorefs
mkdocs-simple-plugin
mkdocstrings-python
jupyter_contrib_nbextensions
notebook==6.4.12

[extras]
matplotlib
jupyter
jupyterlab
tqdm
chainconsumer
numpyro
dLuxToliman
scikit-learn
