sklearn
pytorch_lightning
scikit-learn
torchvision
torchmetrics
torch