torch>=2.0
codecarbon>=3.1.0
numpy
