开源项目

DeepSelect发布高性能TopK实现

deepseek-ai/DeepSelect

AI 基础设施模型服务框架

概述

DeepSelect 提供 TopK 算子实现,即从一组得分中快速选择排名最高的若干项。 DeepSelect面向DeepSeek稀疏注意力和采样中的特定TopK负载,项目报告相对torch.topk取得2—20倍局部算子加速。 DeepSelect是DeepSeek稀疏注意力及采样器所用TopK算子的高性能实现,适用于DeepSeek V3.2、V4与V4.1。项目报告相对原生torch.topk取得2—20倍算子加速。