Skip to content

Positions & normalization · QK-Norm

QK-Norm

RMSNorm applied to queries and keys before attention to keep logits well-behaved.

Normalising queries and keys (usually before RoPE) prevents attention logits from growing too large, a common source of loss spikes in large-scale training.

OLMo 2 and Qwen3 helped make it standard; it is now a default ingredient in many recipes.

Training stability comparison associated with OLMo 2's normalization changes. Figure from Sebastian Raschka’s explainer.

Adoption over time

Models in the atlas using QK-Norm, per half-year of release.

Loading

Used by

40

models in the atlas

Earliest: OLMo 2 (7B) (Nov 25, 2024)

Models using QK-Norm

  • Architecture diagram of LFM3 (1.7B)

    Liquid AI, released 2026-10-27. Dense hybrid decoder, 1.7B parameters, 256K context. Attention: ShortConv plus gated GQA with QK-Norm and sliding-window/global attention.

  • Architecture diagram of MiniMax M3.1-Flash (428B)

    MiniMax, released 2026-10-19. Sparse MoE decoder, 428B total, 23B active, 1M context. Attention: GQA with QK-Norm and MiniMax Sparse Attention.

  • Architecture diagram of Recursive Flint (40B-A2.5B)

    Recursive, released 2026-10-18. Sparse MoE decoder, 40B total, 2.5B active, 256K context. Attention: GQA with QK-Norm, learned attention sinks, and sliding-window/global attention.

  • Architecture diagram of Reflection Beam (501B)

    Reflection, released 2026-10-16. Sparse MoE decoder, 501B total, 23B active, 1M context. Attention: Gated GQA with QK-Norm and sliding-window/global attention.

  • Architecture diagram of Dory1.0 (4B)

    NVIDIA, released 2026-10-16. Dense decoder, 4B parameters, 256K context. Attention: Gated GQA; 3:1 sliding-window/full, RoPE.

  • Architecture diagram of Muse Spark Open (816B-A42B)

    Meta, released 2026-10-15. Sparse MoE decoder, 816B total, 42B active, 1M context. Attention: Gated GQA with scaleless QK-Norm, local-only RoPE, global NoPE, and 3:1 sliding-window/global attention.

  • Architecture diagram of Kolibri-1 (78B)

    Aleph Alpha, released 2026-10-03. Sparse MoE decoder, 78B total, 3.5B active, 256K context. Attention: GQA with QK-Norm and 4:1 sliding-window/global attention.

  • Architecture diagram of Step-5 Preview (600B)

    StepFun, released 2026-09-20. Sparse MoE decoder, 600B total, 27B active, 1M context. Attention: Gated GQA with QK-Norm and 3:1 sliding-window/sparse attention.

  • Architecture diagram of Qwen3.8-Flash-Next (125B-A6B)

    Alibaba Cloud, released 2026-08-26. Hybrid MoE decoder, 125B total, 6B active, 256K context. Attention: 3:1 Gated DeltaNet and Qwen Sparse Attention (QSA).

  • Architecture diagram of Ornith 1.5 (35B-A3B)

    Ornith, released 2026-08-23. Hybrid MoE decoder, 35B total, 3B active, 256K context. Attention: 3:1 Gated DeltaNet and Gated Attention.

  • Architecture diagram of Qwen3.8 (27B)

    Alibaba Cloud, released 2026-08-14. Dense hybrid decoder, 27B parameters, 256K context. Attention: 3:1 Gated DeltaNet and Gated Attention.

  • Architecture diagram of Muse Glimmer (30B)

    Meta, released 2026-08-10. Dense decoder, 30B parameters, 128K context. Attention: Gated GQA with scaleless QK-Norm, local-only RoPE, global NoPE, and 3:1 sliding-window/global attention.

Search the atlas

Jump to a model, concept or page