ElemedexEnciclopedia de bolsillo de arquitecturas de IA
N.º 052

Híbrido SSM (Jamba, Qwen3.5)

SSM–attention hybrid · hybrid linear attention · Jamba · Nemotron-H · Qwen3-Next · Qwen3.5

Quimera

Tiene el cuerpo de una serpiente y, cada tanto, un ojo que mira para atrás. Siete de cada ocho vértebras digieren el texto sin volver sobre él; la octava lo relee entero cuando hay que citar al pie de la letra. Corre casi como Mamba, recuerda casi como un transformer, y discute con los dos sobre quién es su verdadero padre.

Altura
Jamba: 32 (4 bloques × 8: 28 de Mamba, 4 de atención)
Peso
Jamba: ≈ 104 GB (bf16); cabe en una GPU de 80 GB con pesos int8
Parámetros
Jamba: 52 B totales · 12 B activos por token (Qwen3.5: 397 B · 17 B)
Contexto
Jamba: 256K tokens, con una caché KV de 4 GB (Mixtral: 32 GB)
Datos de entrenamiento
Jamba: web, libros y código hasta marzo de 2024; tamaño no divulgado
Año
2024
Generación
6 · Asistentes, razonamiento y agentes
Creadores
Lieber et al. (AI21 Labs, Jamba, 2024); De et al. (Google DeepMind, Griffin, 2024); NVIDIA (Nemotron-H, 2025); Qwen Team (Alibaba, Qwen3-Next, 2025, and Qwen3.5, 2026)
Clase
Arquitectura

Estadísticas base

Total
430

Escalas ordinales y aproximadas, pensadas para comparar especies entre sí; los números duros están en la ficha.

Debilidades

  • La caché KV de las capas de atención sigue creciendo con el contexto: más despacio, pero crece.
  • Cuántas capas de atención poner, y dónde, se decide con ablaciones en modelos chicos, no con una teoría.
  • Rinde mejor con núcleos de cómputo especiales para las capas recurrentes, que fichas como las de Jamba y Qwen3-Next recomiendan instalar aparte.
  • Casi todas las cifras de la familia vienen de los informes de los propios laboratorios.

Hábitat

Nació en AI21 Labs (Jamba, 2024) y en los laboratorios de NVIDIA y Google DeepMind. Hoy vive en modelos de pesos abiertos como Qwen3.5 y Nemotron 3, sobre todo donde se leen documentos enteros o se razona durante miles de tokens.

Movimientos

Habilidades

La proporción
Una capa de atención cada ocho (Jamba) o cada cuatro (Qwen3.5): pocas alcanzan para recuperar del contexto, y la caché KV se achica en la misma proporción.
Dos memorias
La mayoría de las capas lleva un estado comprimido de tamaño fijo; unas pocas guardan cada token tal cual y pueden volver a leerlo.
Sin posiciones
Jamba no usa codificación posicional explícita: la recurrencia de las capas Mamba ya sabe qué vino antes.

Anatomía

Bloque Jamba (8 capas)×4una de cada ochoTokensEmbeddings (sin posiciones)Mamba + MLPMamba + MoE (top-2 de 16)Mamba + MLPMamba + MoEAtención (GQA) + MLPMamba + MoEMamba + MLPMamba + MoESiguiente tokenContexto de 256K tokens28 capas: estado fijo4 capas: caché KV completaCaché: 4 GB (Mixtral: 32)JambaQué guarda al generar

Izquierda: el bloque de Jamba, repetido cuatro veces. De sus ocho capas, siete son Mamba y una (la quinta) es de atención; las redes feedforward alternan entre un MLP común y una mezcla de 16 expertos que usa dos por token. Cada capa lleva su conexión residual y su normalización (omitidas). Derecha: al generar, las capas Mamba cargan un estado que no crece; solo las cuatro de atención guardan claves y valores de cada token del contexto.

Cadena evolutiva

  • la mayoría de las capas sigue siendo recurrente; vuelven unas pocas de atención para recordar043 Mamba
  • conserva un puñado de capas de atención (una de cada ocho en Jamba)025 Transformer
052 Híbrido SSM (Jamba, Qwen3.5)

Un híbrido es un modelo de lenguaje que intercala dos clases de capa. La mayoría son de tiempo lineal y llevan la historia comprimida en un estado de tamaño fijo: un modelo de espacio de estados (SSM) como Mamba o una «atención lineal». Unas pocas son de atención completa, como las del Transformer. En los modelos más recientes suele sumarse una mezcla de expertos (MoE), como en Mezcla de expertos (Mixtral). El ejemplar que dio a conocer la familia es Jamba, de AI21 Labs (marzo de 2024): 52.000 millones de parámetros, 12.000 activos por token, y solo 4 de sus 32 capas son de atención (Lieber et al., 2024). Sus autores lo presentaron como el primer híbrido de atención y SSM apto para producción (Lieber et al., 2024, sec. 1); Griffin, de Google DeepMind, se había publicado un mes antes (De et al., 2024).

Por qué mezclar

Cada progenitor falla donde el otro acierta. La atención guarda claves y valores de todos los tokens anteriores en la caché KV, que crece con el contexto: con 256.000 tokens, Mixtral necesita 32 GB de caché; Jamba, 4 GB (Lieber et al., 2024, sec. 2). El estado fijo, en cambio, no puede guardarlo todo: Jelassi et al. (2024) probaron que un transformer de dos capas copia cadenas de longitud exponencial en su tamaño, mientras que todo modelo de estado fijo, sea un SSM o una atención lineal, queda limitado por el tamaño del estado. En las ablaciones de Jamba, con modelos de 1.300 millones de parámetros, el Mamba puro a menudo no respetaba el formato de los ejemplos del prompt (contestaba «Very Good» donde había que elegir entre «Positive» y «Negative»); el híbrido sí, y en sus capas de atención los autores encontraron cabezas que se comportan como cabezas de inducción, evidencia que ellos mismos llaman anecdótica (Lieber et al., 2024, sec. 6.2). Dao y Gu conjeturan una división del trabajo: las capas SSM hacen de transformación general de secuencias y la atención, de mecanismo de recuperación, que evita comprimir todo el contexto en el estado (Dao y Gu, 2024, sec. 9.2.3).

La familia

Proporciones según los informes de cada modelo (De et al., 2024; Waleffe et al., 2024; NVIDIA, 2025a; MiniMax, 2025; Qwen Team, 2025; Qwen Team, 2026; Kimi Team, 2025; NVIDIA, 2025b):

Modelo Año Capa de estado fijo Estado : atención MoE
Jamba 2024 Mamba 7 : 1 sí
Griffin 2024 recurrencia lineal con compuertas 2 : 1 (atención local) no
Mamba-2-Hybrid 8B 2024 Mamba-2 24 : 4 no
MiniMax-Text-01 2025 atención lineal 7 : 1 sí
Nemotron-H 56B 2025 Mamba-2 54 : 10 no
Qwen3-Next, Qwen3.5 2025–26 Gated DeltaNet 3 : 1 sí
Kimi Linear 2025 Gated DeltaNet refinado 3 : 1 sí
Nemotron 3 2025 Mamba-2 pocas capas de atención sí

Gated DeltaNet combina dos maneras de escribir en la memoria: una compuerta que borra rápido y la regla delta, que corrige solo la entrada asociada a una clave (S. Yang et al., 2025). El buque insignia de Qwen3.5 tiene 397.000 millones de parámetros, 17.000 activos, y pesos abiertos (Qwen Team, 2026).

Lo que se discute

Cuántas. Las ablaciones, casi siempre en modelos chicos, ubican el óptimo entre el 8 y el 25 % de capas de atención, aunque no todos cuentan igual: cerca del 10 % para Dao y Gu (2024, sec. 9.2.3); cerca del 8 % para Waleffe et al. (2024), que incluyen las MLP en el total (2024, sec. 4.1); tres capas lineales por cada una de atención para Kimi Linear (Kimi Team, 2025, sec. 5.2). Jamba no encontró diferencias sustanciales entre 1 : 3 y 1 : 7 (Lieber et al., 2024, sec. 6.1).

Dónde. Repartidas de manera pareja, sin patrón mejor a la vista, y con una capa recurrente al principio, que codifica la posición y vuelve innecesaria la codificación posicional (Waleffe et al., 2024, sec. 4.1). No todos prescinden de ella: las capas de atención de Qwen3-Next y Qwen3.5 aplican posiciones rotatorias (RoPE) a un cuarto de las dimensiones de cada cabeza (Qwen Team, 2025; Qwen Team, 2026).

Si son dos cosas. Mamba-2 nació de mostrar que los SSM y ciertas variantes de la atención son dos descomposiciones de una misma clase de matrices (Dao y Gu, 2024), y el artículo de Gated DeltaNet presenta a Mamba-2 como una atención lineal con decaimiento (S. Yang et al., 2025, sec. 2.1). La oposición real no sería recurrencia contra atención, sino comprimir contra no comprimir. Kimi Linear afirma ser el primer híbrido que supera a la atención completa en una comparación justa (Kimi Team, 2025): es una afirmación del propio laboratorio.

Fuentes