numpy
scipy
scikit-learn
torch==0.3.1
matplotlib
