clearml
pytorch-lightning >= 1.6.0
torch
torchvision