torch>=1.4
numpy
scipy
