2024
FlashAttention-3: Fast and Accurate Attention with Asynchrony and Low-precision
NeurIPS 2024spotlight
Attention, as a core layer of the ubiquitous Transformer architecture, is the bottleneck for large language models and long-context applications. elaborated an approach to speed up attention on GPUs through minimizing memory reads/writes. However, it has yet to take advantage of new capabilities pre…