numpy
scipy
torch
torchvision
torchmetrics
scikit-learn
