AI21 Labs logo

Jamba

AI21 LabsOpen WeightsPending Human Review

Jamba is an AI21 language-model release. The original Jamba combines attention and Mamba state-space layers with sparse experts. It provides a 256K context with a smaller attention cache than a pure Transformer, aiming at efficient long-document language modeling. The original checkpoint is a base completion model; instruction-tuned derivatives and later versions have different terms. Long context does not guarantee complete retrieval or factual correctness, and deployed throughput depends on hardware, precision, and prompt length.

2024-03-28
52B total, 12B active
Hybrid Mamba state-space / Transformer; MoE in larger tiers
Apache-2.0

Specifications

Parameters
52B total, 12B active
Architecture
Hybrid Mamba state-space / Transformer; MoE in larger tiers
License
Apache-2.0
Context Window
262,144 tokens
Type
text
Modalities
text

Benchmark Scores

Advanced Specifications

Model Family
Jamba
API Access
Available
Chat Interface
Not Available
Variants
Original v0.1

Capabilities & Limitations

Capabilities
text completionlong context
Known Limitations
Long-input quality requires task evaluationLicense and weight footprint vary by tier
Notable Use Cases
enterprise question answeringdocument assistantsprivate deployment

Related Models