numpy>=1.21
matplotlib>=3.4
pyvista>=0.38
psutil>=5.9.0
ipython>=7.0.0
sympy>=1.10

[dev]
pytest
ruff

[jax]
jax>=0.4.14
jaxlib>=0.4.14
diffrax>=0.6.2

[notebooks]
ipywidgets
ipympl
notebook

[torch]
torch>=2.1
