clearml
pytorch_lightning >= 1.1.2
torch
torchvision