torch>=1.8
numpy
scipy
