setuptools>=52.0.0
torch>=1.7.1
