torch>=2.9
numpy
