torch==1.11.0
torch-geometric==2.1.0
e3nn==0.5.8
plotly
periodictable
py3Dmol
gdown
wandb

[dev]
pytest
black
isort

[gpu]
torch==1.11.0+cu113
torch-geometric==2.1.0
