AI & Machine Learning

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.

Publisher / source
Orchestra-Research/AI-Research-SKILLs (found on skills.sh, an awesome list)
Upstream record
Orchestra-Research/AI-Research-SKILLs
Version
Version 3
Published
Publication review
Approved for publication
Content fingerprint
e0c1da83766332e0f3db4bcca954c3a70be399632758dcec2680aa82e236438d