jax>=0.4
jaxlib
equinox

[dev]
ruff
pytest
voltax[docs,examples]

[docs]
mkdocs<2,>=1.6
mkdocs-material>=9.5
mkdocstrings[python]>=0.26

[examples]
matplotlib
optax
blackjax
scikit-learn
