MoonEP 开源:用省显存反传设计优化 MoE 训练

SeunghyunSEO7 · x · 2026-07-28

MoonEP 已经开源,帖子强调它在 MoE 训练里采用了 sonic backprop 相关数学设计来降低 activation memory。

配图展示的是 memory-efficient permutation 的实现思路:通过重排 router / permutation 流程,减少反向传播时需要保留的张量,在尽量不改功能计算的前提下压低激活显存占用。作者还提到这项工作与 NVIDIA Megatron 团队有关,并特意点到 Tri Dao 一贯擅长的省显存设计。

所属事件:Moonshot 开源 MoonEP:MoE 专家并行通信与负载均衡库(9 条相关)→

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →