torch>=2.7.1
numpy
