numpy
scipy
torch>=1.7
typing_extensions
