torch>=2.1
numpy
