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

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