typing-extensions
torch
torchvision
scikit-learn
