DeepSeek开源周再添新成员,推出专为MoE和EP设计的通信库DeepEP。首个用于MoE模型训练和推理的开源EP通信库,提供高吞吐量和低延迟的all-to-all GPU内核,并支持FP8运算。开源协议采用宽松的MIT许可证。
DeepEP在GitHub上线后反响热烈,用户纷纷点赞。该库在性能上表现优异,具备高效的all-to-all通信、NVLink和RDMA支持、高吞吐量和低延迟内核,以及原生FP8调度支持和灵活的GPU资源控制。
DeepEP分为两种内核:一种针对高吞吐量任务,另一种则注重低延迟。前者适用于训练和推理预填充任务,后者则优化了延迟敏感型解码场景。DeepEP团队在H800上进行了测试,确保其在不同场景下的性能。
使用DeepEP需配备特定硬件和软件环境,包括Hopper GPU、CUDA 12.3及以上版本、PyTorch 2.1及以上版本等。网络配置方面,DeepEP已在InfiniBand网络上全面测试,并兼容RoCE。
为优化性能,DeepEP建议启用自适应路由,但在网络负载较轻时使用静态路由。团队还发现一条PTX指令能提升性能,但可能影响其他平台,用户可根据情况调整。
DeepSeek在GitHub上新开库https://github.com/deepseek-ai/open-infra-index,预计开源周内容均与AI基础设施相关。
原文链接
本文链接:https://kx.umi6.com/article/14246.html
转载请注明文章出处
相关推荐
.png)
换一换
已节省数百万GPU小时!字节再砍MoE训练成本,核心代码全开源
2025-03-11 16:36:46
榨干每一块GPU,DeepSeek开源第二天,送上降本增效神器
2025-02-25 16:02:59
DeepSeek开源第一个用于混合专家模型训练和推理的高效专家并行通信库
2025-02-25 12:00:43
DeepSeek开源第二弹,为MoE和EP量身定制的通信库!暂和英伟达显卡绑定
2025-02-25 11:49:28
豆包提出全新稀疏模型架构 UltraMem,推理成本较 MoE 最高可降 83%
2025-02-12 13:32:45
对话中国工程院院士郑纬民:DeepSeek,究竟厉害在哪里
2025-01-27 12:28:38
对话蜜度CTO刘益东:价格战、MoE......中国AI大模型下一步“卷”什么?
2024-07-05 20:47:52
推理成本比MoE直降83%!字节最新大模型架构入围ICLR 2025
2025-02-12 12:26:20
训练MoE足足提速70%!华为只用了3招
2025-06-03 15:58:16
MoE 高效训练的 A/B 面:与魔鬼做交易,用「显存」换「性能」
2024-06-07 18:18:13
MoE那么大,几段代码就能稳稳推理 | 开源
2025-07-02 18:09:04
DeepSeek前实习生魔改MoE,用迭代机制把内存需求砍了42%,团队:“免费午餐”优化方法
2025-03-08 11:43:54
大模型的中场战事:CoE or MoE ?
2024-08-03 11:48:38
516 文章
171892 浏览
24小时热文
更多

-
2025-09-08 20:04:22
-
2025-09-08 20:03:26
-
2025-09-08 19:03:01