numpy
sympy
pyvista
scipy
tqdm

[dev]
pytest
pre-commit
nbsphinx
pydata-sphinx-theme
jupyterlab

[jax]
jax[cuda]
diffrax
optax

[torch]
torch
torchdiffeq
