torch
torch-geometric
pytorch-lightning
ase
numpy
jaxtyping

[dev]
notebook
ruff
bumpver

[doc]
sphinx
furo
nbsphinx
sphinx-autobuild
sphinxext-opengraph
sphinx-copybutton

[publish]
build
twine

[test]
pytest
pytest-cov
