torch>=1.4.0
numpy
scipy
matplotlib
sklearn
