Cursor's MoK Optimizes GPU Pipeline, Boosts MoE Training by 41%

icon MarsBit
Share
AI summary iconSummary
Cursor has open-sourced Mixture-of-Kittens (MoK) to enhance MoE model training. MoK integrates GPU data transfer and computation into a single kernel, reducing idle time. On 512 GB300 GPUs, training throughput increased by 41%. The tool is now used in Composer training across thousands of GPUs and is available under the Apache 2.0 license. It supports NVIDIA Blackwell GPUs. On-chain data reveals growing interest in AI infrastructure, with on-chain analysis highlighting efficiency improvements in large-scale training environments.

According to Beating Monitor, Cursor has open-sourced Mixture-of-Kittens (MoK) to accelerate MoE large model training. MoK combines previously separate GPU data transfers and computations into a single kernel—a low-level program running directly on the GPU. MoE splits models into numerous “experts,” activating only a subset at a time. Since these experts are distributed across different GPUs, data must be frequently moved, sometimes consuming over half of the training time. MoK enables GPUs to transfer data and perform computations simultaneously, reducing idle waiting. In real-world training using 512 GB300 GPUs, MoK improved overall throughput by 41%. When tested independently on MoE layers, it achieved up to 2.37x faster performance than public alternatives. MoK has already been deployed in Cursor’s Composer training across tens of thousands of GPUs and is now open-sourced under Apache 2.0. Currently, it supports only NVIDIA Blackwell GPUs and is primarily designed for institutions with GB200, GB300, or NVL72 clusters.

Disclaimer: The information on this page may have been obtained from third parties and does not necessarily reflect the views or opinions of KuCoin. This content is provided for general informational purposes only, without any representation or warranty of any kind, nor shall it be construed as financial or investment advice. KuCoin shall not be liable for any errors or omissions, or for any outcomes resulting from the use of this information. Investments in digital assets can be risky. Please carefully evaluate the risks of a product and your risk tolerance based on your own financial circumstances. For more information, please refer to our Terms of Use and Risk Disclosure.