[None][refactor] Refactor Kimi K3 MLP - #17312

Merged

jiaganc merged 9 commits into

NVIDIA:mainfrom

jiaganc:codex/kimi-k3-gated-mlp-tp

Aug 18, 2026

Commits on Aug 17, 2026

[None][refactor] Refactor Kimi K3 MLP
jiaganc
committed
2d45284
fix(torch): address Kimi K3 MLP review
jiaganc
committed
5d1e88f
test(torch): avoid TP workspace allocation
jiaganc
committed
1831032
test(torch): validate Kimi K3 MoE output
jiaganc
committed
6a32747
test(torch): use portable Kimi router path
jiaganc
committed
864f74e
test(torch): validate Kimi shared MLP construction
jiaganc
committed
12ad447
refactor(torch): remove unused Kimi RMSNorm flag
jiaganc
committed
58bbab4
fix(torch): address Kimi K3 review feedback
jiaganc
committed
1f49d44
test(torch): select Kimi MoE backend explicitly
jiaganc
committed
6311675