numpy
torchmetrics
torch>=1.1.0
accelerate
