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

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

作者: 发表时间:2025-12-04 3:22:29
西安市公开信息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 资源。

相关文章