Our models are offered on various platforms to accommodate a diverse range of user needs. We continually innovate, introducing new features and enhancing our models.

Here is a detailed breakdown of which models are available on each platform:

Model / Platform	AI21 Platform	Amazon Bedrock	Amazon SageMaker	Microsoft Azure	LlamaIndex	LangChain
Jamba Instruct	✅	✅	❌	✅	✅	✅
Jurassic Ultra	✅	✅	✅	❌	❌	✅
Jurassic Mid	✅️	✅	✅️	❌	❌	✅
Jurassic Light	✅️	❌	✅️	❌	❌	❌
Contextual Answers (In-prompt)	✅️	❌️	✅️	❌	❌	✅
Contextual Answers (Document library)	✅️	❌️	❌️	❌	❌	❌
Document library management	✅️	❌	❌	❌	❌	❌
Text Segmentation	✅️	❌	❌	❌	❌	✅
Paraphrase	✅️	❌️	✅️	❌	❌	❌
Grammatical Error Corrections	✅️	❌️	✅️	❌	❌	❌
Text Improvements	✅️	❌️	❌	❌	❌	❌
Summarize	✅️	❌️	✅️	❌	❌	❌

Platform details

AI21 Platform

AI21 Platform consists of the REST endpoints and the Python SDK.

Foundation Model	Context Window
Jamba Instruct	512K tokens
Jurassic Ultra	8192 tokens
Jurassic Mid	8192 tokens
Jurassic Light	8192 tokens

Task-Specific Model	Input Limit
Paraphrase	500 characters
Grammatical Error Corrections	500 characters
Text Improvements	500 characters
Summarize	50K characters
Contextual Answers (In-prompt)	50K characters
Contextual Answers (Document library)	1 GB
Text Segmentation	100K characters

Amazon

Amazon SageMaker

Foundation Model	Context Window
Jurassic Ultra	See here
Jurassic Mid	See here
Jurassic Light	See here

Task-Specific Model	Input Limit	Notes
Contextual Answers (In-prompt)	See here
Summarize	See here	`Focused Summary` is only available on AI21 Studio
Paraphrase	500 characters
Grammatical Error Correction	500 characters

Amazon Bedrock

These models are available on Amazon Bedrock.

Foundation Model	Version	Context Window
Jamba Instruct		256K
Jurassic Ultra	2.2.000	8192 tokens
Jurassic Mid	2.2.000	8192 tokens

Microsoft Azure

Jamba Instruct is available in Azure AI Studio and the Azure AI model catalog.

LlamaIndex

AI21's latest foundation model is available on LlamaIndex.

See the full instructions on GitHub
Try out the code in a Colab notebook.

LangChain

Some AI21 models and tools are available on LangChain.

Overview of AI21 objects in LangChain. Includes links to LangChain wrapper objects.
Readme for AI21 on LangChain. Includes usage snippets.

LangChain supports the following AI21 components:

Model or tool	LangChain object	Notes
Jamba Instruct	`ChatAI21`	Some object parameters are not supported by Jamba. See the Jamba Instruct documentation to learn which parameters are supported by this model.
Jurassic	`AI21LLM`	Jurassic is deprecated; use Jamba Instruct instead.
Contextual Answers (In-prompt)	`AI21ContextualAnswers`	Implements the in-prompt contextual answer endpoint.
Text Segmentation	`AI21SemanticTextSplitter`	Implements the Text Segmentation endpoint.
Embeddings	`AI21Embeddings`	Implements the Embeddings endpoint.