torch
torchvision
PIL