@samsja19:通过在MoE层融合计算与通信,我们显著加速了训练栈,并添加了反向…

X AI KOLs Timeline 工具

摘要

Prime Intellect 让 DeepSeek 的 MegaMoE 内核变得可训练,通过在 MoE 层融合计算与通信,在 Blackwell 和 Hopper GPU 上实现了比 TorchTitan 最高快 55% 的 MoE 训练性能,并支持 BF16、MXFP8 和 FP8 精度。

我们通过在 MoE 层融合计算与通信,显著加速了训练栈,并为 DeepSeek MegaMoE 添加了反向传播内核 支持 Blackwell 和 Hopper GPU 上的 BF16、MXFP8 和 FP8
查看原文
查看缓存全文

缓存时间: 2026/10/29 18:03

通过在 MoE 层中融合计算与通信,并为 deepseek MegaMoE 添加反向传播 kernel,我们显著提升了训练栈的速度

支持 Blackwell 和 Hopper GPU 上的 BF16、MXFP8 和 FP8 格式

Prime Intellect (@PrimeIntellect): 介绍 Prime Mega MoE。

我们让 DeepSeek 的 Mega MoE kernel 变得可训练了。在两个 Qwen3 模型上的结果:

• MoE 训练速度比 TorchTitan 快最多 55% • 使用 prime-rl 进行托管训练快最多 34%

相似文章

深入微软前沿预训练技术栈(18 分钟阅读)

TLDR AI

微软公布了其最新的 MoE 基础模型 MAI-Base-2(80B 激活参数 / 2.8T 总参数),在自研评测中超越所有开源权重模型,预训练 FLOPs 效率比 DeepSeek-V4-Pro-Base 高 4-12 倍,使用 16k 张 GB200 GPU 在 38 天内完成训练。