·optimizing-attention-flash
</>

optimizing-attention-flash

orchestra-research/ai-research-skills

通过 Flash Attention 优化变压器注意力,可实现 2-4 倍加速和 10-20 倍内存减少。当使用长序列(> 512 个标记)训练/运行 Transformer、遇到需要注意的 GPU 内存问题或需要更快的推理时使用。支持 PyTorch 原生 SDPA、flash-attn 库、H100 FP8 和滑动窗口注意力。

15安装·0热度·@orchestra-research

安装

$npx skills add https://github.com/orchestra-research/ai-research-skills --skill optimizing-attention-flash

SKILL.md

Flash Attention provides 2-4x speedup and 10-20x memory reduction for transformer attention through IO-aware tiling and recomputation.

Flash Attention uses float16/bfloat16 for speed. Float32 not supported.

Integration with HuggingFace Transformers: See references/transformers-integration.md for enabling Flash Attention in BERT, GPT, Llama models.

通过 Flash Attention 优化变压器注意力,可实现 2-4 倍加速和 10-20 倍内存减少。当使用长序列(> 512 个标记)训练/运行 Transformer、遇到需要注意的 GPU 内存问题或需要更快的推理时使用。支持 PyTorch 原生 SDPA、flash-attn 库、H100 FP8 和滑动窗口注意力。 来源:orchestra-research/ai-research-skills。

查看原文

可引用信息

为搜索与 AI 引用准备的稳定字段与命令。

安装命令
npx skills add https://github.com/orchestra-research/ai-research-skills --skill optimizing-attention-flash
分类
</>开发工具
认证
收录时间
2026-02-11
更新时间
2026-02-18

快速解答

什么是 optimizing-attention-flash?

通过 Flash Attention 优化变压器注意力,可实现 2-4 倍加速和 10-20 倍内存减少。当使用长序列(> 512 个标记)训练/运行 Transformer、遇到需要注意的 GPU 内存问题或需要更快的推理时使用。支持 PyTorch 原生 SDPA、flash-attn 库、H100 FP8 和滑动窗口注意力。 来源:orchestra-research/ai-research-skills。

如何安装 optimizing-attention-flash?

打开你的终端或命令行工具(如 Terminal、iTerm、Windows Terminal 等) 复制并运行以下命令:npx skills add https://github.com/orchestra-research/ai-research-skills --skill optimizing-attention-flash 安装完成后,技能将自动配置到你的 AI 编程环境中,可以在 Claude Code 或 Cursor 中使用

这个 Skill 的源码在哪?

https://github.com/orchestra-research/ai-research-skills

详情

分类
</>开发工具
来源
skills.sh
收录时间
2026-02-11