Skip to content

By playing, you agree to uListen's Terms and Privacy Policy and YouTube's Terms of Service.

Kimi AIKimi AI

How We Scaled Kimi K2.5 | Zhilin Yang's full GTC 2026 Keynote

If you're curious about the "how" behind scaling Kimi's latest models, this is the session you can't miss:)

Mar 21, 202639mWatch on YouTube

Summaries, chapters, transcripts, and FAQs on this page are made by uListen, not YouTube.

CHAPTERS

Chapter summaries

  1. Why open models matter: democratizing intelligence beyond black boxes

    The speaker frames the company’s pursuit of better open models as a way to democratize access to intelligence. They emphasize deployability anywhere and full access to model weights, and point to open models rapidly closing the gap with proprietary systems.

    • •Open models can be deployed locally or in the cloud
    • •Access to model weights is contrasted with black-box usage
    • •Open models are described as quickly reaching frontier performance
    • •Goal is making intelligence accessible globally
    • •The talk will focus on how to make open models not just open, but great
  2. Three scaling dimensions: token efficiency, long context, and agent swarms

    The speaker introduces scaling as a major driver of AI progress, then broadens the notion of scaling into three dimensions. They outline token efficiency (shifting the scaling curve left), increasing context length, and orchestrating multiple agents to expand task capacity.

    • •Token efficiency aims to achieve lower loss with the same number of training tokens
    • •Longer context length improves accuracy at predicting tokens at given positions
    • •Agent swarms introduce a paradigm where multiple agents solve subtasks in parallel
    • •Token efficiency is linked to stronger priors for agent search via agent I/O
    • •Vision for combining these: swarms of long-context agents with strong priors
  3. Token efficiency isn’t just cost: it raises the intelligence ceiling in a data-limited world

    The speaker revisits classical scaling laws and argues token efficiency is more than infrastructure efficiency. With high-quality data being limited, improvements in token efficiency effectively act like having more high-quality tokens, pushing the upper bound of achievable intelligence.

    • •Classical scaling law: scale tokens, parameters, and compute to reduce loss
    • •Token efficiency is positioned as increasing the achievable intelligence, not only saving compute
    • •Example framing: doubling token efficiency makes a fixed dataset act like twice as many tokens
    • •High-quality data is described as limited, motivating efficiency work
    • •This motivates significant investment in token-efficiency techniques
  4. Muon optimizer: second-order updates, scalable implementation, and measured gains

    The speaker introduces Muon as a second-order optimizer that transforms gradient updates so entries are orthogonal. They describe techniques used to make it scale for large LLM training and present results showing broad improvements when replacing AdamW with Muon under equal training budgets.

    • •Muon is described as transforming each gradient update to make entries orthogonal
    • •Claimed outcome: about 2× token-efficiency improvement when implemented properly
    • •Two scaling techniques highlighted: weight decay and consistent RMS updates vs Adam
    • •Adjustable coefficients are used so update RMS is comparable to Adam
    • •A distributed implementation partitions optimizer state across the data-parallel group for memory efficiency
  5. Training instability at trillion-parameter scale and the QK clip solution

    When scaling Muon to a 1-trillion-parameter model, the speaker reports instability: exploding max logits and loss divergence. They introduce QK clip, which computes max logits per attention head and scales query/key projections to constrain values, stabilizing training without harming loss reduction.

    • •Observed failure mode: max logits exploding beyond typical ranges and loss diverging
    • •QK clip computes max logits per attention head during the forward pass
    • •A dividing factor is applied to key and query projections to clip/constrain values
    • •Training curves overlap before/after clipping, while max logits become constrained
    • •Technique is reported as enabling successful scaling in their K2 training to 1T parameters
  6. Why long context matters: transformers improve across the whole context unlike LSTMs

    The speaker motivates long-context scaling using a comparison between transformers and LSTMs. They highlight that transformer loss continues to improve deeper into the context, whereas LSTM performance saturates, and connect long context to harder agent-like tasks requiring long trajectories.

    • •Transformers achieve lower loss than LSTMs under comparable budgets in the cited comparison
    • •Transformer loss is described as continuously dropping as token index increases
    • •LSTM loss is described as saturating after a certain context length
    • •Long context is framed as essential for harder tasks and long-running agent trajectories
    • •Motivation: build architectures that scale context efficiently while reducing per-token loss at large indices
  7. Kimi Linear and Delta Attention: expressive linear attention mixed with full attention

    The speaker introduces Kimi Linear, featuring Kimi Delta Attention, to scale to longer contexts efficiently. They describe mixing linear attention layers with full attention layers in a one-to-three ratio to balance long-context capability and efficient implementation.

    • •New architecture introduced: Kimi Linear with Kimi Delta Attention
    • •Kimi Delta Attention is described as improving a gated delta rule variant via improved recurrent memory
    • •Linear attention layers are mixed with full attention layers at a one-to-three ratio
    • •Goal is efficient long-context scaling while maintaining strong performance
    • •Sets up detailed discussion of fine-grained decay and chunkwise computation
  8. Fine-grained decay and exact chunkwise reformulation to make linear attention scalable

    The speaker explains a limitation of linear attention with a single global decay factor and proposes a channel-wise decay (diagonal matrix) to improve expressivity. They then describe infrastructure challenges introduced by this change and an exact reformulation that enables parallel implementation without approximation.

    • •Problem with global decay: either forget almost everything or retain too much indiscriminately
    • •Solution: replace scalar decay with a diagonal matrix to control per-channel decay rates
    • •Different channels can retain long-range information or quickly forget to refresh
    • •Chunkwise formulation is used to parallelize on GPUs
    • •Equation is rewritten with matrix inversion and cumulative decay for efficiency, stated to be mathematically equivalent (not an approximation)
  9. Results for Kimi Linear: better short- and long-context performance plus efficiency at huge lengths

    The speaker presents comparative results showing Kimi Linear outperforming other variants on both short-context and long-context tasks while being efficient. They claim it can outperform full attention across short input, long input, and long output tasks, especially as context scales to very large lengths.

    • •Reported improvements on a short-context task (MMANU) versus MLA and GDN
    • •Reported improvements on a long-context task (Ruler) versus other variants
    • •Efficiency advantages become larger at very long contexts (e.g., one million tokens and beyond)
    • •Claimed to be the first architecture described as outperforming full attention across the board
    • •Positioned as a key pillar of the long-context scaling dimension
  10. Agent swarms: orchestrating parallel sub-agents to increase task capacity and reduce time

    The speaker introduces an agent swarm paradigm with an orchestrator that spawns sub-agents, assigns tasks, and aggregates results iteratively. They argue this reduces execution time for complex tasks compared to single-agent approaches and enables scaling inputs, outputs, actions, and orchestration.

    • •Architecture: a main agent/orchestrator spawns and coordinates sub-agents
    • •Process is iterative: assign tasks, collect results, and refine
    • •Analogy to human organizations with different roles coordinating toward a goal
    • •Framing: for higher task complexity, agent swarms reduce execution time versus single agents
    • •Scales along multiple axes: parallel reading/downloading sources, writing long outputs in parallel, performing actions across tasks, and orchestrating aggregation
  11. Training agent swarms with three rewards: instantiation, finish, and outcome

    The speaker describes new objective terms used to train the agent swarm system. They define rewards to encourage parallelism early, ensure spawned subtasks actually complete, and ultimately optimize for successful task outcomes, with reward weights decayed over training.

    • •Instantiation reward discourages “serial collapse” into single-agent execution and encourages parallelism early
    • •Weight on instantiation reward can be decayed as parallel behavior is learned
    • •Finish reward addresses spawned subtasks that never complete and discourages meaningless task spawning
    • •Finish reward weight is also decayed over training
    • •Outcome reward measures whether the overall task is completed; all three are combined in an RL system and require supporting infrastructure for parallel execution and multiple rewards
  12. Putting it together: Kimi K2.5 release, emergent multimodal coding/design, and stable pretraining

    The speaker connects the three scaling dimensions to the creation and release of Kimi K2.5. They describe observed emergent capabilities that combine visual understanding with coding and highlight an unusually smooth, stable training curve over very large token counts on NVIDIA H100 infrastructure.

    • •K2.5 is presented as combining gains from token efficiency, long context architecture, and agent swarms
    • •A demo video is referenced showing capabilities like reading a video and producing a website that replicates or style-transfers it
    • •Speaker attributes these capabilities to successful and stable pretraining
    • •Training curve is described as highly stable with no loss spikes, including after introducing Muon
    • •Training details mentioned: over fifteen trillion tokens for the base model and additional training beyond that; trained on NVIDIA H100 clusters with stated node configuration and NVLink connectivity
  13. Native joint vision-text training: early fusion, cross-modal gains, and zero-vision SFT

    The speaker claims K2.5 is the first open model with native joint vision and text capabilities via early fusion from the start of training, contrasting with adding vision later. They describe findings that vision-only RL can improve text tasks, and a strong text base can support vision performance with zero vision SFT data when combined with joint RL.

    • •Approach: early fusion of vision and text tokens from day one, contrasted with late fusion after text pretraining
    • •Preliminary experiments are said to show early fusion outperforming late fusion
    • •Claim: properly trained, vision and text can enhance each other instead of degrading text performance
    • •Observation: vision-only RL (vision tasks without text tasks) improves performance on text tasks
    • •Training recipe described: zero-vision SFT (only text SFT data) followed by joint RL over text and vision, yielding strong vision results without vision SFT data
  14. Attention Residue: ‘attention rotated by 90 degrees’ as a next-generation depth mechanism

    The speaker previews a new architecture called Attention Residue, motivated by applying ideas from temporal recurrence to network depth. They reinterpret residual connections as analogous to recurrent mechanisms and propose using attention over previous layer states (plus a block variant) to improve efficiency and performance.

    • •New work introduced: Attention Residue, presented as a sneak peek at a future architecture direction
    • •Motivation: apply techniques inspired by temporal modeling to the depth dimension
    • •Residual connections are described as analogous to a recurrent mechanism across layers
    • •Core idea: instead of using only the previous layer, attend over all previous hidden states to form the current layer output
    • •Efficiency variant: block attention residue attends over block outputs while using standard residuals within blocks to reduce overhead
  15. Measured improvements, research mindset shift, and closing summary on continued scaling

    The speaker reports efficiency and benchmark improvements from Attention Residue and reflects on why older ideas can still be improved with today’s scaling and evaluation rigor. They close by reiterating continued scaling across multiple dimensions and collaboration with the open-source community.

    • •Reported token-efficiency improvement: 24% and consistently lower validation loss vs baseline curve
    • •Largest improvements are said to appear on coding, math, and reasoning-heavy benchmarks named by the speaker
    • •Speaker argues modern resources and benchmark suites enable more rigorous conclusions than a decade ago
    • •They list a progression of revisiting core components: optimizers, attention mechanisms, and residual connections
    • •Conclusion: continue scaling along token efficiency, context length, and beyond (agent swarms not the end) while advancing with the open-source community

Get more out of YouTube videos.

Read along on your phone, search it beside the video, or ask your favorite AI app what was said.