torch
torchvision
pillow
kornia
