NVIDIA Transformer Engine 加速 JAX 无丢弃 MoE 训练,DeepSeek-V3 吞吐提升 10.4 倍
NVIDIA Transformer Engine 结合 JAX,在 NVIDIA GB200 上把 DeepSeek-V3 的 MoE 训练吞吐从 103 TFLOPS/GPU 提升到 1,068 TFLOPS/GPU,达 10.4 倍。其分组 GEMM 内核可在单次调用中处理各专家不等的 token 数量,实现无丢弃 MoE;NCCL EP 融合 dispatch 与 combine 并去重 token 以降低网络流量。在 GB300 NVL72 上训练 DeepSeek-V3 671B 时,该方案在 1,024 块 GPU 下保持 97% 扩展效率。
匹配字段标题
