torch>=1.3
numpy
scipy
