中国人工智能公司DeepSeek在其“开源周”的第三天宣布推出了一款名为DeepGEMM的开源库，该库支持FP8通用矩阵乘法（GEMM）。这一工具专为密集型和混合专家（MoE）矩阵运算设计，旨在为DeepSeek V3和R1模型的训练与推理提供强大的支持。官方消息通过X平台发布后，迅速引发了技术社区的广泛关注和热烈讨论。
根据DeepSeek官方账号发布的帖子，DeepGEMM在NVIDIA Hopper GPU上能够实现高达1350+ TFLOPS的FP8计算性能。尽管其核心逻辑仅包含约300行代码，但该库在大多数矩阵尺寸上的表现甚至超越了经过专家调优的内核，展现了极高的效率和简洁性。DeepGEMM无需复杂的依赖，采用即时编译（Just-In-Time）技术，支持密集布局和两种MoE布局，设计上“像教程一样干净”，便于开发者学习和使用。
X用户@TechBitDaily评论道：“DeepGEMM的发布是DeepSeek开源周的一大亮点，其FP8性能和简洁设计令人印象深刻。”另一位用户@AIObserverCN则指出，该库在支持MoE模型的高效训练方面具有显著优势，可能会推动AI社区在Hopper架构上的进一步创新。
作为开源周的一部分，DeepGEMM的发布延续了DeepSeek推动AI技术透明化和社区协作的承诺。此前，该公司已在开源周的前两天发布了FlashMLA和DeepEP工具，分别聚焦于快速语言模型架构和专家并行通信。此次DeepGEMM的亮相，进一步展示了DeepSeek在AI基础设施建设上的技术实力。业内人士认为，这款库不仅将提升DeepSeek自家模型的性能，也为全球开发者提供了一个高效、易用的矩阵运算工具，未来应用前景广阔。用户现可通过GitHub获取DeepGEMM，探索其在AI训练与推理中的潜力。
项目地址：https://github.com/deepseek-ai/DeepGEMM