torch>=2.4.0
numpy
