教程 / 实战普通
NVIDIA Transformer Engine 如何用 JAX 加速 Dropless MoE 训练
原始标题:Accelerating Dropless MoE Training in JAX with NVIDIA Transformer Engine
内容摘要
NVIDIA 在 JAX 中用 Transformer Engine 优化 Dropless MoE 训练,在 NVIDIA GB200 上跑 DeepSeek-V3 时,未优化基线仅 103 TFLOPS/GPU、GPU 间通信占累计 kernel 时间 84%,经针对性 kernel 优化后升至 1,068 TFLOPS/GPU,提升 10.4 倍。
内容分类AI 教程与实战
内容层级普通情报
发布时间(北京时间)
本站收录时间(北京时间)
信息来源NVIDIA Technical Blog:Agentic AI / Generative AI
站内情报编号intel-59246eaf5f4809e69a6b6a16