setuptools
wheel
numpy
typing_extensions
torch