Use PyTorch with the SageMaker Python SDK — sagemaker 2.235.1 documentation
sagemaker.readthedocs.io