torch>=1.6
numpy
