torch>=2.8
numpy
