numpy
setuptools
torch>=1.4
higher
matplotlib
