DeepSeek第二炸!开源首个用于MoE模型训练通信库

DeepSeek第二炸!开源首个用于MoE模型训练通信库

作者: 发表时间:2025-12-07 4:10:36
大庆市防洪app 陇南市教研app 濮阳市工商信息app 安庆市劳动局app 佛山市防灾信息app 宁德市最新新闻app 淮安市工程造价app 宜昌市税收公开app 江门市天气监控中心app 牡丹江市水务app 平凉市土地申报app 湘潭市工商局app 东营市水利app 咸阳市防灾信息app 乐山市第四中学app 厦门市森林消防app 贺州市警务app 日喀则市历史记录app 孝感市劳动局app 新乡市事业单位app 厦门市政务服务app 克拉玛依市税收公开app 金昌市电视台广播app 青岛市水利app 锦州市残联救助app 吴忠市特殊家庭关爱协会app 临沭县新闻中心app 和硕县应急管理app 响水县最新新闻app 商河县未成年保护协会app 肃宁县消费协会app 布拖县便民app 锦屏县土地局app 独山县助农app 石楼县公路管理app 绥江县住房保障app 融水苗族自治县新闻中心app 淅川县出口管理app 沈丘县应急管理app 精河县法律服务app 曲周县妇联救助app 乐东黎族自治县税务局app 东光县国土信息app 社旗县教育信息app 共和县劳动局app 正定县第五小学app

DeepSeek 的“开源周”活动今日已经来到第二天,今天发布的是首个开源的用于 MoE 模型训练和推理的 EP 通信库 ——DeepEP。

本站

官方表示其具备如下特征:

高效优化的全到全通信方式

支持节点内外通信,兼容 NVLink 和 RDMA 技术

提供高吞吐量的内核,提升训练和推理前期填充效率

提供低延迟内核,优化推理解码速度

完全支持 FP8 数据格式调度

提供灵活的 GPU 资源管理,支持计算与通信的重叠执行

本站

据介绍,DeepEP 是一款专为混合专家(MoE)和专家并行(EP)设计的通信库,提供了高吞吐量和低延迟的 all-to-all GPU 内核,常用于 MoE 派发和合并操作。该库还支持低精度计算,包括 FP8。

为了与 DeepSeek-V3 论文中提出的组限制门控算法兼容,DeepEP 提供了一些针对不对称带宽转发优化的内核,比如将数据从 NVLink 域转发到 RDMA 域。这些优化的内核能够提供高吞吐量,适合用于训练和推理的预填充任务,同时支持 SM(流式多处理器)数量控制。

对于延迟敏感型的推理解码任务,DeepEP 提供了一套低延迟内核,采用纯 RDMA 技术以最大程度减少延迟。此外,该库还采用了一种基于 Hook 的通信与计算重叠方法,不会占用任何 SM 资源。

相关文章