numpy
setuptools
torch>=1.4
higher
