您的当前位置:首页>科技咨询>资讯详情

DeepSeek开源DeepGEMM代码库

发表于:2025-02-26 19:00:05 浏览:20次 发布者: 网易互联网

北京商报讯(记者 魏蔚)2月26日,Deepseek宣布,开源DeepGEMM代码库。DeepGEMM 是一个专为简洁高效的 FP8 通用矩阵乘法(GEMM)设计的库,具有细粒度缩放功能,如 DeepSeek-V3 中所提出。它支持普通和混合专家(MoE)分组的 GEMM。该库采用 CUDA 编写,在安装过程中无需编译,通过使用轻量级的即时编译(JIT)模块在运行时编译所有内核。FP8GEMM库,为V3/R1训练和推理提供支持。