scipy
numpy
torch>=1.11.0
