jax
jaxtyping
numpy
scipy
interpax
diffrax>=0.7.1
beartype
vmecpp
safetensors
matplotlib

[docs]
zensical
mkdocstrings-python
mkdocs-gallery
pyvista[jupyter]

[plot3d]
pyvista[jupyter]

[test]
pytest
mypy
ruff
pre-commit
