numpy
pandas
scipy
matplotlib
meshio
fenics-basix
gmsh
equinox
optax
scikit-sparse

[cuda12]
jax[cuda12]
nvidia-cudss-cu12
nvidia-cublas-cu12
nvidia-cuda-runtime-cu12
nvidia-cudnn-cu12
nvidia-cuda-nvcc-cu12
nanobind
scikit-build-core>=0.5
ninja
cmake

[cuda13]
nvidia-cudss-cu13
nvidia-cublas
nvidia-cuda-runtime
nvidia-cudnn-cu13
nvidia-cuda-nvcc
nanobind
scikit-build-core>=0.5
ninja
cmake

[dev]
pytest
pytest-cov
black
ruff
mypy

[docs]
sphinx
sphinx-rtd-theme

[jax]
jax[cuda13]
