torch
torchmetrics
torchvision
tqdm
