numpy
ase
tqdm
torch!=1.9.0,<=1.12,>=1.8
e3nn<0.6.0,>=0.3.5
pyyaml
torch-runstats>=0.2.0
torch-ema>=0.3.0
scikit_learn<=1.0.1

[:python_version < "3.7"]
contextlib2
contextvars

[:python_version < "3.8"]
typing_extensions
