torch
pytorch-lightning
ase==3.22.1
numpy
rich
dacite
e3nn
scikit-learn
locache
load-atoms
wandb
data2objects

[dev]
notebook
ruff
bumpver
pyright
sphinx-autobuild

[docs]
sphinx
furo
nbsphinx
sphinxext-opengraph
sphinx-copybutton
sphinx-design

[publish]
build
twine

[test]
pytest
pytest-cov
pyright
