numpy
matplotlib
jax
diffrax

[dev]
pytest
pytest-cov
ruff
mypy
build

[docs]
mkdocs-material
mkdocstrings[python]
pymdown-extensions
