torch
torchvision
joblib
scikit-learn==1.2.2