Built-in Algorithms

Amazon SageMaker provides implementations of some common machine learning algorithms optimized for GPU architecture and massive datasets.

• Tabular
  • AutoGluon
  • CatBoost
  • Factorization Machines
  • K-Nearest Neighbors
  • LightGBM
  • LinearLearner
  • TabTransformer
  • XGBoost
  • Object2Vec
• Text
  • Blazing Text
  • LDA
  • NTM
  • Sequence-to-Sequence
  • Text Classification - TensorFlow
  • Sentence Pair Classification - TensorFlow
  • Sentence Pair Classification - HuggingFace
  • Question Answering - PyTorch
  • Named Entity Recognition - HuggingFace
  • Text Summarization - HuggingFace
  • Text Generation - HuggingFace
  • Machine Translation - HuggingFace
  • Text Embedding - TensorFlow, MxNet
• Time-series
  • Deep AR Forecasting
• Unsupervised
  • IP Insights
  • K-means
  • PCA
  • Random Cut Forest
• Vision
  • Image Classification - MxNet
  • Image Classification - PyTorch
  • Image Classification - TensorFlow
  • Object Detection - MxNet GluonCV
  • Object Detection - MxNet
  • Object Detection - PyTorch
  • Object Detection - TensorFlow
  • Semantic Segmentation - MxNet GluonCV
  • Semantic Segmentation - MxNet
  • Instance Segmentation - MXNet
  • Image Embedding - TensorFlow
• Other
  • Amazon Estimators