typing_extensions
matfree>=0.6.1

[cpu]
jax[cpu]

[doc]
jupytext
jupyter
matplotlib
diffeqzoo
tueplots
tqdm
optax
blackjax[progress]>=1.0.0
diffrax
numba
mkdocs<2.0.0
mkdocs-material<9.7.0
mkdocs-exclude
mkdocstrings-python
mkdocstrings
mkdocs-jupyter
mkdocs-awesome-pages-plugin

[format-and-lint]
pre-commit

[test]
pytest<9.0
pytest-xdist
pytest-cases<3.10
diffeqzoo
diffrax
equinox
