Available foundation models
Amazon SageMaker JumpStart offers state-of-the-art, built-in publicly available and proprietary foundation models to customize and integrate into your generative AI workflows.
Publicly available foundation models
Amazon SageMaker JumpStart onboards and maintains open source foundation models from third-party sources. To get started with one of these publicly available models, see JumpStart foundation model usage or explore one of the available Example notebooks. In a given example notebook for a publicly available model, try switching out the model ID to experiment with different models within the same model family.
For more information on model IDs and resources on deploying publicly available JumpStart foundation models with the SageMaker Python SDK, see Use foundation models with the SageMaker Python SDK.
By definition, foundation models are adaptable to many downstream tasks. Foundation models are trained on massive amounts of general domain data and the same model can be implemented or customized for multiple use cases. When choosing your foundation model, start with defining a specific task, such as text generation or image generation.
Publicly available time series forecasting models
Time series forecasting models are designed to analyze and make predictions on sequential data over time. These models can be applied to various domains such as finance, weather forecasting, or energy demand forecasting. The Chronos models are tailored for time series forecasting tasks, enabling accurate predictions based on historical data patterns.
| Model Name | Model ID | Model Source | Fine-tunable |
|---|---|---|---|
Chronos T5 Small |
autogluon-forecasting-chronos-t5-small |
Amazon |
No |
Chronos T5 Base |
autogluon-forecasting-chronos-t5-base |
Amazon |
No |
Chronos T5 Large |
autogluon-forecasting-chronos-t5-large |
Amazon |
No |
| Chronos-Bolt Small | autogluon-forecasting-chronos-bolt-small |
Amazon |
No |
| Chronos-Bolt Base | autogluon-forecasting-chronos-bolt-base |
Amazon |
No |
Publicly available text generation models
Text generation foundation models can be used for a variety of downstream tasks, including text summarization, text classification, question answering, long-form content generation, short-form copywriting, information extraction, and more.
To explore the latest text generation JumpStart foundation models, use the
Text Generation filter on the Getting started with Amazon SageMaker JumpStart
Publicly available image generation models
JumpStart provides a wide variety of Stable Diffusion image generation foundation models including base models from Stability AI as well as pre-trained models for specific text-to-image tasks from Hugging Face. If you need to fine-tune your text-to-image foundation model, you can use Stable Diffusion 2.1 base from Stability AI. If you want to explore models that are already trained on specific art styles, you can explore one of the many third-party models from Hugging Face directly in the Amazon SageMaker Studio UI or SageMaker Studio Classic UI.
To explore the latest image generation JumpStart foundation models, use the
Text to Image filter on the Getting started with Amazon SageMaker JumpStart
Proprietary foundation models
Amazon SageMaker JumpStart provides access to proprietary foundation models from third-party
providers such as AI21 Labs
To get started with one of these proprietary models, see JumpStart foundation model usage. To use a proprietary foundation model, you must first subscribe to the model in Amazon Web Services Marketplace. After subscribing to the model, locate the foundation model in Studio or SageMaker Studio Classic. For more information, see SageMaker JumpStart pretrained models.
To explore the latest proprietary foundation models for a variety of use cases,
see Getting started with Amazon SageMaker JumpStart