Frameworks The SageMaker Python SDK supports managed training and inference for a variety of machine learning frameworks: mxnet/index chainer/index huggingface/index pytorch/index rl/index sklearn/index sparkml/index tensorflow/index xgboost/index djl/index