torch>=1.12
e3nn==0.4.4
numpy<2.0
opt_einsum
ase
torch-ema
prettytable
matscipy
h5py
torchmetrics
python-hostlist
configargparse
GitPython
pyYAML
tqdm
cuequivariance-torch
matplotlib
pandas

[cueq-cuda-11]
cuequivariance-ops-torch-cu11

[cueq-cuda-12]
cuequivariance-ops-torch-cu12

[dev]
black
isort
mypy
pre-commit
pytest
pytest-benchmark
pylint

[fpsample]
fpsample

[schedulefree]
schedulefree

[wandb]
wandb
