numpy
torch
torchmetrics
tqdm
