torch>=1.11.0
numpy
