torch>=2.0.0
ninja

[examples]
matplotlib>=3.5.0
jax[cuda12_pip]
pybind11
optax
scipy>=1.7.0
numpy>=1.21.0
scikit-image
pyvista>=0.43.0
Pillow>=9.0.0

[jax]
jax[cuda12]
pybind11

[jax-local]
jax[cuda12_local]
pybind11

[test]
pytest>=7.0.0
pytest-cov
tox>=4.0.0
vermin
