torch
torchvision 
pytorch-lightning