Agents

NEW paper from Apple. Interesting idea: 'Attention to Mamba'. The paper introduces a two-stage recipe for cross-architecture distillation fr…

NEW paper from Apple. Interesting idea: 'Attention to Mamba'. The paper introduces a two-stage recipe for cross-architecture distillation from Transformers into Mamba. Naive distillation collapses tea

DGX agentx-post
agentsdair-ai--x

NEW paper from Apple. Interesting idea: "Attention to Mamba". The paper introduces a two-stage recipe for cross-architecture distillation from Transformers into Mamba. Naive distillation collapses teacher performance. Their trick: first distill the transformer into a linearized-attention student using a kernel adaptation, then transfer that student into a pure Mamba with no attention blocks. On a 1B model trained on 10B tokens, the Mamba student hits 14.11 perplexity against a 13.86 Pythia-1B teacher, nearly matching quality at linear-time inference cost. If you can reliably convert trained transformers into state-space models without retraining from scratch, the entire open-weights ecosystem becomes cheaper to serve at long context. This is the kind of quiet infrastructure work that decides which architectures actually get deployed in agent stacks. Paper: https://arxiv.org/abs/2604.14191 Learn to build effective AI agents in our academy: https://academy.dair.ai/

Related

Source: DAIR.AI (X) | 2026-04-18

Loading related sources…