Armory
Source
Browse
Skills

optimizing-attention-flash

Optimizes transformer attention with Flash Attention for 2-4x speedup and 10-20x memory reduction. Use when training/running transformers with long sequences (>512 tokens), encountering GPU memory issues with attention, or need faster inference. Supports PyTorch native SDPA, flash-attn library, H100 FP8, and sliding window attention.

Score
Unranked
Evidence
No signals yet
Last commit
Not known
Listed

Install

armory install optimizing-attention-flash --cli claude

writes the skill into.claude/skills/optimizing-attention-flash/SKILL.mdListed as compatible

Configuration
# fetches the source and writes it to:
.claude/skills/optimizing-attention-flash/SKILL.md

Needs the armory CLI · not on npm yet, build it from cli/ in the repository

What it is

Optimizes transformer attention with Flash Attention for 2-4x speedup and 10-20x memory reduction. Use when training/running transformers with long sequences (>512 tokens), encountering GPU memory issues with attention, or need faster inference. Supports PyTorch native SDPA, flash-attn library, H100 FP8, and sliding window attention.

When to use it

Optimizes transformer attention with Flash Attention for 2-4x speedup and 10-20x memory reduction. Use when training/running transformers with long sequences (>512 tokens), encountering GPU memory issues with attention, or need faster inference. Supports PyTorch native SDPA, flash-attn library, H100 FP8, and sliding window attention.

How to install / invoke

# Copy the skill into your .claude/skills/ directory
curl -sL https://raw.githubusercontent.com/davila7/claude-code-templates/main/cli-tool/components/skills/ai-research/optimization-flash-attention/SKILL.md -o .claude/skills/optimization-flash-attention/SKILL.md

Notes

Extracted from davila7/claude-code-templates, optimization-flash-attention category.