setuptools
torch
