setuptools
torch
numpy
