torch
torchvision
datasets
huggingface_hub
optuna
torchmetrics
scikit-learn