Frameworks

The SageMaker Python SDK supports managed training and inference for a variety of machine learning frameworks:

• Apache MXNet
  • Use MXNet with the SageMaker Python SDK
  • MXNet Classes
• Chainer
  • Using Chainer with the SageMaker Python SDK
  • Chainer
• Hugging Face
  • HuggingFace
  • Use Hugging Face with the SageMaker Python SDK
• PyTorch
  • PyTorch
  • Use PyTorch with the SageMaker Python SDK
• Reinforcement Learning
  • Using Reinforcement Learning with the SageMaker Python SDK
  • RLEstimator
• Scikit-Learn
  • Using Scikit-learn with the SageMaker Python SDK
  • Scikit Learn
• SparkML Serving
  • SparkML Serving
• TensorFlow
  • Use TensorFlow with the SageMaker Python SDK
  • Deploying to TensorFlow Serving Endpoints
  • Upgrade from Legacy TensorFlow Support
  • TensorFlow
• XGBoost
  • Use XGBoost with the SageMaker Python SDK
  • XGBoost Classes for Open Source Version