torch>=2.7
numpy
