setuptools
wheel
numpy
typing_extensions
torch==1.12
parameterized