torch
torchvision
matplotlib
tqdm
wandb
