pytorch
torchvision
kornia
scipy
tqdm
wandb