numpy
torch
scipy
einops
psutil

[dev]
pytest>=7.4.0
pytest-cov>=3.10.0
mypy
pre-commit
ruff>=0.4.0
typos
torchdiffeq

[docs]
mkdocs-material>=9.4.0
mkdocstrings[python]>=0.22.0
mkdocs-gen-files>=0.5.0
mkdocs-literate-nav>=0.6.0
pillow>=10.0.0
cairosvg>=2.7.1
