numpy
scipy
torch>=1.11
