torch
pynvml
matplotlib
numpy
torchprofile
codecarbon
