torch>=1.8
e3nn==0.4.4
numpy
ase
