jax
jaxtyping
numpy
plum-dispatch
scipy

[dev]
pre-commit
pyright
jupyterlab
isort>=7.0.0
ruff

[docs]
mkdocs-material
mkdocstrings[python]

[examples]
exponax
matplotlib
optax
tueplots

[interop]
skerch>=1.2
torch

[test]
pytest<9,>=8
pytest-cov
pytest-cases
hypothesis
tox-uv
