KTransformers-FineTune: CPU-GPU co-compute boosts MoE fine-tuning up to 21.2x

jiqizhixin · x · 2026-10-25

Tsinghua's MADSys Lab and KV-Cache Tech present KTransformers-FineTune (KT-FT), addressing the bottleneck where fine-tuning large MoE models streams expert weights over PCIe from CPU memory. Since MoE's sparse compute finishes too fast for GPU compute to mask the weight transfer, KT-FT instead lets the CPU directly participate in MoE fine-tuning computation alongside the GPU. Result: 2.5x–21.2x fine-tuning throughput vs. traditional CPU-GPU weight offloading, with an ecosystem integration into the open-source LlamaFactory project. Paper authors span Tsinghua, Zhejiang, Beihang, and DeepSeek-AI.

Original post →

More from Infra

Infra channel →