matplotlib
lightning
torchmetrics
torchvision
torch-geometric
kornia
scipy
scikit-image
