torch
torchmetrics
tqdm
