Ai21 Labs Large Language Models Discoverable In Amazon Sagemaker Studio

AI21 Labs’ Large Language Models Discoverable in Amazon SageMaker Studio | AI21

Skip to Main Menu

Skip to Main Content

Skip to Footer

Back to Blog

-->

Back to Blog

Our large language models (LLMs), which were previously accessible to a select few, can now be discovered in Amazon SageMaker Studio via Amazon SageMaker JumpStart.

Amazon SageMaker JumpStart is a machine learning (ML) hub that provides access to a wide range of publicly available foundation models along with built-in algorithms and end-to-end solution templates, all of which can be seamlessly integrated within the ML development workflow.

Through SageMaker, customers can host our models on a private instance using Amazon’s Virtual Private Cloud (VPC) services, providing a secure environment so that all related data strictly remains there.

SageMaker also offers flexibility for customers to select the servers for running AI21 models, providing an optimal trade-off between cost and throughput. SageMaker users can customize deployments according to their specific needs, ensuring efficiency in both performance and costs.

Since initially announcing our availability on SageMaker, we’ve upgraded our foundation models to Jurassic-2 and have also added task-specific models to be accessible to all users.

AI21 Models Available in SageMaker Studio

Jurassic-2 Foundation Models

Jurassic-2 Ultra

Jurassic-2 Mid

Jurassic-2 Light

Jurassic-2 models are among the most advanced LLMs in the world, ranked for accuracy and reliability, according to Stanford’s Holistic Evaluation of Language Models. They are capable of versatile, human-like text generation (both long and short-form) and solving complex tasks like answering questions, text classification, among other things.

Task-Specific Models

Contextual Answers

Summarize

Paraphrase

Grammatical Error Correction (GEC)

Task-specific models (APIs) are highly-optimized models that empower businesses to integrate specialized language tasks producing exceptionally high quality into their applications.

By tapping into Jurassic-2 and task-specific models, you can leverage the latest and greatest generative AI capabilities to securely build and easily deploy generative AI-powered solutions, apps, and/or services.

Through SageMaker, we look forward to collaborating with more businesses to develop advanced textual generative AI applications.

Interested in learning more about AI21 Labs on SageMaker? Let’s talk.

Discover more

Jun 25, 2026

Token spend isn’t going down. You need more than naive routing to manage it

Labs in Front

-->

Jun 24, 2026

Tipping the scales: Merging weak agents into a state-of-the-art deep researcher

Labs in Front

-->

Jun 4, 2026

First scale, then enrich: How the right execution strategy helped us reach state-of-the-art on SWE-rebench