matplotlib
numpy
scipy
torch