torch>=1.7.1
pytorch_lightning
torchvision
lightning-bolts
