torch>=2.0.0
torchdiffeq>=0.2.4
numpy
scipy
matplotlib

[data]
jax>=0.4.12
exponax>=0.1.0

[docs]
mkdocs-material>=9.5
mkdocstrings[python]>=0.24
mkdocs-jupyter>=0.24
