torch
torchvision
sagemaker>=2.156.0,<3