numpy
sympy
pyvista
scipy
tqdm

[dev]
pytest
pre-commit
nbsphinx
pydata-sphinx-theme
jupyterlab
nbval
black
isort
nbconvert==7.17.1

[jax]
jax>=0.8.0
diffrax
optax

[torch]
torch
torchdiffeq
