numpy<2.0.0
torch>=2.2.0
torch-geometric>=2.5.0
e3nn==0.5.8
torch-scatter
torch-sparse
torch-spline-conv
torch-cluster
plotly
periodictable
py3Dmol
gdown
wandb

[dev]
pytest
black
isort

[gpu]
torch>=2.2.0
