submitit
tqdm
wandb

[torch]
torch
torchvision
