jax
jaxlib
zodiax>=0.4.1

[docs]
matplotlib
jupyter
jupyterlab
tqdm
chainconsumer
numpyro
dLuxToliman
scikit-learn
mkdocs
mkdocs-jupyter
mkdocs-same-dir
mkdocs-autorefs
mkdocs-simple-plugin
mkdocstrings-python
jupyter_contrib_nbextensions

[tests]
pytest
