inference/research
verifiedhigh · 0.90updated 2026-10-03

Kimi-K2

moe 32/1026B active

Fields

id
models/kimi-k2
Family
kimi-k2
Vendor
Moonshot AI
Release year
2025
Architecture
moe
Params (B)
1,026
Active params (B)
32
Experts
384
Experts per token
8
Shared expert
yes
Hidden size
7,168
Layers
61
KV heads
64
Head dim
128
GQA ratio
1
Max context
131,072
Context scaling
YaRN (factor 32, original 4096, beta 1/1)
Context scaling method
yarn
Activation
swiglu
Attention variant
MLA, kv_lora_rank 512, q_lora_rank 1536
KV LoRA rank (MLA)
512
QK rope head dim
64
QK nope head dim
128
V head dim
128
KV cache (bytes/token)
70,272
FLOPs per active token
64,000,000,000
Open weights
yes
License
Modified MIT License (Moonshot, bespoke clauses)
notes
The largest open-weight MoE at its release and the first to exceed 1T total. Vendor: 1T total / 32B activated; safetensors total 1026.41B; our config arithmetic gives 1026.2B / 32.7B, matching to ~2%. The brief's '1T' is confirmed and then some. kv_cache_bytes_per_token = 61 layers x (kv_lora_rank 512 + qk_rope_head_dim 64) x 2 bytes = 61 x 576 x 2 = 70,272 B/token (68.6 KiB) - identical to DeepSeek-V3, because K2 is DeepSeek-V3-shaped: same 61 layers, same MLA geometry, same hidden 7168. It is V3 scaled in experts, not in depth: 384 routed + 1 shared (V3's 256), 8 per token, but first_k_dense_replace 1 instead of 3, and num_key_value_heads 64 instead of 128. qk_nope_head_dim 128, qk_rope_head_dim 64, v_head_dim 128, q_lora_rank 1536 - all identical to V3. Context: max_position_embeddings 131072 via YaRN factor 32 over an original 4096, rope_theta 50000 (V3 uses 10000) with beta_fast 1.0 / beta_slow 1.0 (V3 uses 32 / 1) - a gentler, more uniform RoPE adjustment. Trained with the Muon optimizer, which is the notable difference from V3 rather than an architectural one. num_nextn_predict_layers 0 - no MTP. Licence is a 'Modified MIT License' (Moonshot, 2025) - commercially usable but with vendor-specific clauses; not plain MIT. Kimi-K2-Thinking is the same config with a different serving mode. SERVICEABILITY JOIN (added 2026-10-03; the architecture x engine matrix this record was missing, with a citation per cell in docs/13-model-compatibility.md): FAMILY=mla. REQUIRED KERNEL CLASS: an MLA decode kernel. MLA is not a variant of the GQA kernel - the cache is a shared low-rank latent (d_c) plus a decoupled RoPE key, not per-head K/V, so an engine that lacks a dedicated MLA decode path cannot serve the class at all. SUPPORTED BY (cited): vLLM native rows DeepseekV2ForCausalLM, DeepseekV3ForCausalLM, DeepseekV32ForCausalLM, DeepseekV4ForCausalLM [[mdl2-vllm-supported-models-2026-10]]; TensorRT-LLM the same four rows plus GlmMoeDsaForCausalLM (GLM-5), Glm4MoeForCausalLM (GLM-4.5/4.6/4.7), KimiK3ForConditionalGeneration and KimiLinearForCausalLM [[mdl2-trtllm-supported-models]]; SGLang generative table names DeepSeek v1/v2/v3/R1 and Kimi K2 as supported [[mdl2-sglang-generative-models]]; llama.cpp reaches MLA through the DEEPSEEK2/DEEPSEEK32/DEEPSEEK4 arch entries, which are in its supports_sm_tensor == false list, i.e. no tensor-core matmul path for the attention itself [[mdl2-llamacpp-arch-class-registry]]; ik-llama-cpp records MLA as a feature that landed there first and was later upstreamed [[ik-llama-cpp]].ENGINE KERNEL CONSTRAINTS THAT CHANGE WHAT THE ENGINE MUST STORE (this is the important class of fact): (1) FlashInfer, which is SGLang's default attention backend and selectable in vLLM [[flashinfer]], is where the MLA decode kernels live, and they are instantiated per KV cache page-block size: gotcha-sm120-sparse-mla-block-size records vLLM computing compressed-layer pbs = block_size // compress_ratio (= 32 at block_size 64 for a ratio-2 layer) while FlashInfer's SM120 planner hard-codes _PAGE_BLOCK_SIZE = 64, so DeepSeek-V4.1-Flash will not start on 8x RTX PRO 6000 at TP8 at ANY global block size (64 gives pbs=32 and crashes; 128 gives 'No common block size'). Workaround is SM100 datacenter Blackwell or a FlashInfer planner change. (2) FlashMLA's 2026.09.30 release requires SM100/SM103 and CUDA >= 13.1, REMOVED Hopper support and removed V3/V3.2/V4.0, and supports ONLY FP8 and FP4 KV cache - unquantized bfloat16 KV is explicitly not supported [[kern-flashmla-github]]. So the kernel choice decides the KV CACHE FORMAT: to use that decode kernel the engine must store an FP8/FP4 latent with per-token inlined scales, and V4.1's layout is 528 bytes/token (512 of float8_e4m3 over all 512 dims - the 64 RoPE dims are quantized too, no bf16 part - plus 16 bytes of float8_e8m0 scales, one per 32 e4m3 values), with an fp4 variant at 288 bytes/token [[flashmla-decode-fp8-fp4-kv]]. A Hopper box therefore cannot take that kernel and an engine that only offers bf16 MLA KV cannot take it either. (3) Independent of the kernel: TRT-LLM's MLA rows carry two footnotes that constrain serving rather than correctness - chunked prefill for MLA only enables on SM90/SM100/SM103/SM107/SM120, and KV cache reuse for MLA only enables on SM90/100/103/107/120/121 in BF16/FP8 [[mdl2-trtllm-supported-models]]. (4) TRT-LLM states DeepSeek-V4 is Blackwell SM100+ only and Kimi K3 is SM100 family only, with a worked recipe split (DEP16 needs GB300-class per-GPU memory, TEP16 validated on GB200) [[mdl2-trtllm-supported-models]]. ISA CONSEQUENCE: at 68.6 KiB/token MLA moves the bottleneck off KV and onto weight streaming, so MLA-capable engines find their paged-attention machinery largely idle and their GQA-era cache-pressure tuning irrelevant [[mla-latent-attention]]. TRT-LLM's DeepseekV3ForCausalLM row is labelled 'DeepSeek-V3, Kimi-K2' - the same implementation serves both, because they share the MLA geometry in this record [[mdl2-trtllm-supported-models]]. That is the clearest evidence in the repo that MLA is one kernel class rather than one model. CONTEXT_SCALING_METHOD (yarn): config-derived; the existing free text 'YaRN (factor 32, original 4096, beta 1/1)' agrees.

Notes

The largest open-weight MoE at its release and the first to exceed 1T total. Vendor: 1T total / 32B activated; safetensors total 1026.41B; our config arithmetic gives 1026.2B / 32.7B, matching to ~2%. The brief's '1T' is confirmed and then some. kv_cache_bytes_per_token = 61 layers x (kv_lora_rank 512 + qk_rope_head_dim 64) x 2 bytes = 61 x 576 x 2 = 70,272 B/token (68.6 KiB) - identical to DeepSeek-V3, because K2 is DeepSeek-V3-shaped: same 61 layers, same MLA geometry, same hidden 7168. It is V3 scaled in experts, not in depth: 384 routed + 1 shared (V3's 256), 8 per token, but first_k_dense_replace 1 instead of 3, and num_key_value_heads 64 instead of 128. qk_nope_head_dim 128, qk_rope_head_dim 64, v_head_dim 128, q_lora_rank 1536 - all identical to V3. Context: max_position_embeddings 131072 via YaRN factor 32 over an original 4096, rope_theta 50000 (V3 uses 10000) with beta_fast 1.0 / beta_slow 1.0 (V3 uses 32 / 1) - a gentler, more uniform RoPE adjustment. Trained with the Muon optimizer, which is the notable difference from V3 rather than an architectural one. num_nextn_predict_layers 0 - no MTP. Licence is a 'Modified MIT License' (Moonshot, 2025) - commercially usable but with vendor-specific clauses; not plain MIT. Kimi-K2-Thinking is the same config with a different serving mode.

SERVICEABILITY JOIN (added 2026-10-03; the architecture x engine matrix this record was missing, with a citation per cell in docs/13-model-compatibility.md): FAMILY=mla. REQUIRED KERNEL CLASS: an MLA decode kernel. MLA is not a variant of the GQA kernel - the cache is a shared low-rank latent (d_c) plus a decoupled RoPE key, not per-head K/V, so an engine that lacks a dedicated MLA decode path cannot serve the class at all. SUPPORTED BY (cited): vLLM native rows DeepseekV2ForCausalLM, DeepseekV3ForCausalLM, DeepseekV32ForCausalLM, DeepseekV4ForCausalLM [[mdl2-vllm-supported-models-2026-10]]; TensorRT-LLM the same four rows plus GlmMoeDsaForCausalLM (GLM-5), Glm4MoeForCausalLM (GLM-4.5/4.6/4.7), KimiK3ForConditionalGeneration and KimiLinearForCausalLM [[mdl2-trtllm-supported-models]]; SGLang generative table names DeepSeek v1/v2/v3/R1 and Kimi K2 as supported [[mdl2-sglang-generative-models]]; llama.cpp reaches MLA through the DEEPSEEK2/DEEPSEEK32/DEEPSEEK4 arch entries, which are in its supports_sm_tensor == false list, i.e. no tensor-core matmul path for the attention itself [[mdl2-llamacpp-arch-class-registry]]; ik-llama-cpp records MLA as a feature that landed there first and was later upstreamed [[ik-llama-cpp]].ENGINE KERNEL CONSTRAINTS THAT CHANGE WHAT THE ENGINE MUST STORE (this is the important class of fact): (1) FlashInfer, which is SGLang's default attention backend and selectable in vLLM [[flashinfer]], is where the MLA decode kernels live, and they are instantiated per KV cache page-block size: gotcha-sm120-sparse-mla-block-size records vLLM computing compressed-layer pbs = block_size // compress_ratio (= 32 at block_size 64 for a ratio-2 layer) while FlashInfer's SM120 planner hard-codes _PAGE_BLOCK_SIZE = 64, so DeepSeek-V4.1-Flash will not start on 8x RTX PRO 6000 at TP8 at ANY global block size (64 gives pbs=32 and crashes; 128 gives 'No common block size'). Workaround is SM100 datacenter Blackwell or a FlashInfer planner change. (2) FlashMLA's 2026.09.30 release requires SM100/SM103 and CUDA >= 13.1, REMOVED Hopper support and removed V3/V3.2/V4.0, and supports ONLY FP8 and FP4 KV cache - unquantized bfloat16 KV is explicitly not supported [[kern-flashmla-github]]. So the kernel choice decides the KV CACHE FORMAT: to use that decode kernel the engine must store an FP8/FP4 latent with per-token inlined scales, and V4.1's layout is 528 bytes/token (512 of float8_e4m3 over all 512 dims - the 64 RoPE dims are quantized too, no bf16 part - plus 16 bytes of float8_e8m0 scales, one per 32 e4m3 values), with an fp4 variant at 288 bytes/token [[flashmla-decode-fp8-fp4-kv]]. A Hopper box therefore cannot take that kernel and an engine that only offers bf16 MLA KV cannot take it either. (3) Independent of the kernel: TRT-LLM's MLA rows carry two footnotes that constrain serving rather than correctness - chunked prefill for MLA only enables on SM90/SM100/SM103/SM107/SM120, and KV cache reuse for MLA only enables on SM90/100/103/107/120/121 in BF16/FP8 [[mdl2-trtllm-supported-models]]. (4) TRT-LLM states DeepSeek-V4 is Blackwell SM100+ only and Kimi K3 is SM100 family only, with a worked recipe split (DEP16 needs GB300-class per-GPU memory, TEP16 validated on GB200) [[mdl2-trtllm-supported-models]]. ISA CONSEQUENCE: at 68.6 KiB/token MLA moves the bottleneck off KV and onto weight streaming, so MLA-capable engines find their paged-attention machinery largely idle and their GQA-era cache-pressure tuning irrelevant [[mla-latent-attention]]. TRT-LLM's DeepseekV3ForCausalLM row is labelled 'DeepSeek-V3, Kimi-K2' - the same implementation serves both, because they share the MLA geometry in this record [[mdl2-trtllm-supported-models]]. That is the clearest evidence in the repo that MLA is one kernel class rather than one model.

CONTEXT_SCALING_METHOD (yarn): config-derived; the existing free text 'YaRN (factor 32, original 4096, beta 1/1)' agrees.

Cross-references

This record references no other records — it is a leaf. It is cited by 2.