torch
torchvision
pytorch-lightning