pytorch-lightning
torchmetrics
torchvision