lightning
torchmetrics
torchvision
kornia
scipy
scikit-image
