deepseek-ai /DeepSelect

Ускоренные CUDA-вычисления для операции TopK, применяемой в моделях DeepSeek, разработка команды DeepSeek.

DeepSelect — реализация операции TopK (выбор нескольких наибольших значений), используемой в разреженном механизме внимания моделей DeepSeek V3.2 и V4, а также при генерации текста. По сравнению со стандартной функцией torch.topk даёт ускорение от 2 до 20 раз в зависимости от размера данных, но охватывает только конкретные сценарии использования, описанные автором.

Описание автора: DeepSelect: TopK kernels for DeepSeek Sparse Attention (DSA) and Samplers

Cuda★ 358создан 9 сентября 2026форков 22MIT

Открыть на GitHub →

Сколько звёзд прибавилось

Прирост по дням за 3 месяца

17 июня: ≈018 июня: ≈019 июня: ≈020 июня: ≈021 июня: ≈022 июня: ≈023 июня: ≈024 июня: ≈025 июня: ≈026 июня: ≈027 июня: ≈028 июня: ≈029 июня: ≈030 июня: ≈01 июля: ≈02 июля: ≈03 июля: ≈04 июля: ≈05 июля: ≈06 июля: ≈07 июля: ≈08 июля: ≈09 июля: ≈010 июля: ≈011 июля: ≈012 июля: ≈013 июля: ≈014 июля: ≈015 июля: ≈016 июля: ≈017 июля: ≈018 июля: ≈019 июля: ≈020 июля: ≈021 июля: ≈022 июля: ≈023 июля: ≈024 июля: ≈025 июля: ≈026 июля: ≈027 июля: ≈028 июля: ≈029 июля: ≈030 июля: ≈031 июля: ≈01 августа: ≈02 августа: ≈03 августа: ≈04 августа: ≈05 августа: ≈06 августа: ≈07 августа: ≈08 августа: ≈09 августа: ≈010 августа: ≈011 августа: ≈012 августа: ≈013 августа: ≈014 августа: ≈015 августа: ≈016 августа: ≈017 августа: ≈018 августа: ≈019 августа: ≈020 августа: ≈021 августа: ≈022 августа: ≈023 августа: ≈024 августа: ≈025 августа: ≈026 августа: ≈027 августа: ≈028 августа: ≈029 августа: ≈030 августа: ≈031 августа: ≈01 сентября: ≈02 сентября: ≈03 сентября: ≈04 сентября: ≈05 сентября: ≈06 сентября: ≈07 сентября: ≈08 сентября: ≈09 сентября: ≈010 сентября: ≈18511 сентября: ≈6512 сентября: ≈1813 сентября: ≈614 сентября: ≈015 сентября: ≈1217 июняиюльавгсент

Оценка по выборке GH Archive. Бледные столбцы — дни, история за которые ещё загружается.