numpy
matplotlib
torch
scipy
setuptools
