torch
torchvision
scikit-learn
sagemaker>=2.156.0,<3