torch
torchvision
lightning
