submitit
torchmetrics[image]
torchvision
