matplotlib
seaborn
torch
torchmetrics
torchvision
tqdm
