SageMaker Algorithms with Pre-Trained Model Examples by Problem Type

The SageMaker Python SDK provides built-in algorithms with pre-trained models from popular open source model hubs, such as TensorFlow Hub, PyTorch Hub, and Hugging Face. Customers can deploy these pre-trained models as-is, or first fine-tune them on a custom dataset and then deploy to a SageMaker endpoint for inference.

This section provides example notebooks for different ML problem types supported by SageMaker built-in algorithms. Please visit Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK for more documentation.

Example notebooks by problem type

Problem types	Supports inference with pre-trained models	Trainable on a custom dataset	Supported frameworks	Example notebooks
Image classification	Yes	Yes	PyTorch, TensorFlow	Introduction to JumpStart - Image Classification JumpStart - Image Classification Benchmarking and Model Selection
Object detection	Yes	Yes	PyTorch, TensorFlow, MXNet	Introduction to JumpStart - Object Detection
Semantic segmentation	Yes	Yes	MXNet	Introduction to JumpStart - Semantic Segmentation
Instance segmentation	Yes	Yes	MXNet	Introduction to JumpStart - Instance Segmentation
Image embedding	Yes	No	TensorFlow, MXNet	Introduction to JumpStart - Image Embedding
Text classification	Yes	Yes	TensorFlow	Introduction to JumpStart - Text Classification
Sentence pair classification	Yes	Yes	TensorFlow, Hugging Face	Introduction to JumpStart - Sentence Pair Classification
Question answering	Yes	Yes	PyTorch	Introduction to JumpStart - Extractive Question Answering
Named entity recognition	Yes	No	Hugging Face	Introduction to JumpStart - Named Entity Recognition
Text summarization	Yes	No	Hugging Face	Introduction to JumpStart - Text Summarization
Text generation	Yes	No	Hugging Face	Introduction to SageMaker Built-In Algorithms - Text Generation
Machine translation	Yes	No	Hugging Face	Introduction to JumpStart - Machine Translation
Text embedding	Yes	No	TensorFlow, MXNet	Introduction to JumpStart - Text Embedding
Tabular classification	Yes	Yes	LightGBM, CatBoost, XGBoost, AutoGluon-Tabular, TabTransformer, Linear Learner	Tabular classification with Amazon SageMaker LightGBM and CatBoost algorithm Tabular classification with Amazon SageMaker XGBoost and Scikit-learn Linear Learner algorithm Tabular classification with Amazon SageMaker AutoGluon-Tabular algorithm Tabular classification with Amazon SageMaker TabTransformer algorithm
Tabular regression	Yes	Yes	LightGBM, CatBoost, XGBoost, AutoGluon-Tabular, TabTransformer, Linear Learner	Tabular classification with Amazon SageMaker LightGBM and CatBoost algorithm Tabular classification with Amazon SageMaker XGBoost and Scikit-learn Linear Learner algorithm Tabular classification with Amazon SageMaker AutoGluon-Tabular algorithm Tabular classification with Amazon SageMaker TabTransformer algorithm