torch
torchvision
torchmetrics
numpy
scikit-learn
matplotlib
