Building Jamba 3B: the tiny Hybrid Transformer State Space Reasoning Model - Barak Lenz, CTO of AI21

Oct 11, 2025 · 45:05

Barak Lenz, CTO of AI21, presents their Jamba 3B model as a tiny hybrid transformer-state space model that brings long context capabilities to edge devices, and argues enterprises need AI systems like Maestro over standalone models. Lenz explains that the 1:8 ratio of attention to Mamba layers emerged from extensive ablations, with attention placed in the middle of the block working best. The Jamba 3B model uses a 1:12 ratio to maximize efficiency, fitting the same context length as larger models with a fraction of memory. He notes that pure Mamba underperformed on some tasks but hybrid models resolved those deficiencies, and that images quickly become long context problems (4 images can be thousands of tokens). For enterprise, Lenz advocates for model-agnostic orchestration layers treating models as "actions" with statistical properties, enabling continuous learning and cost optimization without vendor lock-in. Drawing from his algo trading background, he compares training frontier models to developing trading algorithms, emphasizing the importance of world-class engineering and avoiding brute force reasoning.

2 episodes