torch
torchmetrics
tqdm