创建

登录 ReadmeX

或

了解我们如何处理你的信息: 隐私政策

ReadmeX

陪你读懂 AI 动态

今天,想先了解什么?

选一份动态总结,或直接聊聊你的问题。

社区和关注动态总结只属于你自己。

资讯

NVIDIA 优化 Ling-3.0-flash 在 DGX Spark 的推理

AI 总结

NVIDIA 技术博客介绍了提升 Ling-3.0-flash 在 DGX Spark 上推理速度的三种方法:Humming MXFP4 MoE 后端、LM head 的在线 FP8 量化,以及基于 NEXTN 的多令牌预测投机解码。分析显示批量为 1 的解码受内存带宽限制,权重读取主导耗时,其中 MoE 计算与输出层投影成本最高。在合成负载上基线为 27.71 ms TPOT、36.09 tok/s,NEXTN MTP 达到 9.72 ms 与 102.88 tok/s;在全部 1,319 道 GSM8K 题目上,三项优化合计使解码吞吐提升 42.5%,平均端到端延迟下降 24.5%,准确率无统计显著变化(96.89% 降至 96.82%)。文章还指出,在线 FP8 LM head 与 MTP 同开时草稿模型需共享整个 LM head 模块,否则会静默回退到比基线更慢的配置。

深度见解

该报告给出一套可在 DGX Spark 上部署的 MoE 推理优化方案与实测数据,并指出静默降级的集成陷阱,同时提醒 185.1% 的合成吞吐增益不能直接外推到其他工作负载。

Generating tokens faster starts with identifying what limits inference performance. In the batch-size-1 evaluation of Ling-3.0-flash from Inclusion AI on NVIDIA DGX Spark, profiling showed that decoding was memory-bound: reading model weights drove execution time. Two operations stood out—the mixture-of-experts (MoE) computation and the output-layer projection that produces scores across the model’s vocabulary.

Those findings pointed to two ways to improve performance: reduce the weight traffic in each decode step and generate more tokens per verification round. This post examines three techniques that address those costs:

  • Humming MXFP4 MoE kernels to improve execution of the routed-expert computation.
  • Online FP8 quantization of the output layer, or LM head, to reduce its weight-read volume.
  • Multi-token prediction (MTP) speculative decoding with NEXTN to draft and verify multiple tokens together.

This post explains how each optimization works, measures its effect, and examines an integration issue that can make the combined configuration slower than the baseline. Ling-3.0-flash serves as the implementation example, with code and deployment instructions in the Ling cookbook.

Model and baseline configuration

The evaluation used Ling-3.0-flash-fp4, a mixture-of-experts model with 124 billion total parameters and 5.1 billion active parameters per token. The model has 42 layers: 35 Kimi Delta Attention (KDA) linear-attention layers and seven multi-head latent attention (MLA) layers. Each MoE layer routes each token to eight of 512 routed experts, alongside one shared expert.

#NVIDIA#Inclusion AI#Ling-3.0-flash#NVIDIA DGX Spark#Humming#FlashInfer#SGLang#NEXTN#GSM8K#FlashInfer CUTLASS

原文约 7 分钟读完阅读原文
群聊

还没有评论,来说说你的看法。