torch>=1.9.0
numpy
scipy
