torch
torchvision
kornia
scipy
tqdm
wandb
