torch>=2.4
numpy
