native-sparse-attention-triton

用Triton实现原生稀疏注意力,长序列推理提速降本。

这是一个基于Triton的高效原生稀疏注意力(NSA)实现。NSA是一种新型注意力机制,通过动态选择重要的键值对,大幅降低长序列场景下的计算和内存开销。该项目解决了标准注意力机制在超长上下文(如128K以上)中计算成本过高的问题,核心能力包括:支持动态稀疏模式、兼容Triton编译器以利用GPU并行能力、提供与PyTorch原生注意力接口类似的API。项目处于早期阶段,代码结构清晰,适合研究者和工程师在长文本处理、多模态推理等场景中集成使用。

开源 free 开源模型
访问官网 ↗ GitHub ↗ 文档 ↗
GitHub 星标 ★ 283
维护状态 低维护
是否开源 是
定价模式 free

项目数据

分类开源模型
开发团队XunhaoLai
所属国家
官网地址
定价模式free
价格说明开源项目,Apache-2.0 许可证,完全免费,可自行部署使用。
访问状态
是否开源是
开源协议Apache-2.0
主要语言Python
技术栈/模型large-language-models,natural-language-processing,sparse-attention
GitHub 星标★ 283
30天Star增速
HF 下载量
上线时间2025-02-24 00:00:00
最近更新2026-09-10 00:00:00
维护状态低维护
中文支持
访问方式
移动端支持
综合评分
收录时间2026-08-09
浏览次数3

使用教程

核心亮点

  • 基于Triton,自动利用GPU并行,性能优于纯PyTorch实现
  • 实现原生稀疏注意力(NSA),动态选择关键token,降低计算量
  • 代码简洁,易于集成到现有Transformer模型中

不足之处

  • 文档/社区待观察
  • 仅支持特定GPU架构,兼容性有限

适用场景

  • 超长文本处理(如文档问答、代码库理解)
  • 大模型推理加速(降低KV cache占用)
  • 多模态长序列任务(如视频理解)

替代项目

FlashAttention、xformers、Sparse Attention Kit

项目介绍

native-sparse-attention-triton 是模型训练领域的开源项目,由 XunhaoLai 开发,2025 年首次发布。

它收录于模型训练分类,该分类目前共有 522 个项目,GitHub 星标数为 283。

项目已超过三个月没有代码更新,维护节奏明显放缓,最近一次代码更新于 2026-09-10。Apache-2.0 许可证,完全免费,可自行部署使用。

它主要面向的使用场景是:超长文本处理(如文档问答、代码库理解)。同类可对比的替代方案包括 FlashAttention、xformers、Sparse Attention Kit。

上一篇:InstructERC

下一篇:ParScale

同类项目推荐

Long-RL 开源

让强化学习轻松驾驭超长序列,训练更稳更快

Long-RL: Scaling RL to Long Sequences (NeurIPS 2025)

★ 729 2026-08-09
TPA 开源

把注意力复杂度从平方降到线性,长序列不再卡顿

[NeurIPS 2025 Spotlight] TPA: Tensor ProducT ATTenTion Transformer (https://arxiv.or···

★ 463 2026-08-09
minimind 开源

两小时造出你的小模型,练手入门不求人

Train a 64M-parameter LLM from scratch in just 2h!

★ 62178 2026-08-09
pruna 开源

一键优化 AI 模型,推理提速又省资源。

Pruna is a model optimization framework built for developers, enabling you to delive···

★ 1286 2026-08-13