numpy
scipy
scikit-learn
torch==1.4.0
matplotlib
