squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add paper arXiv link

Browse Source
This commit is contained in:
Tri Dao 2022-05-29 18:15:43 -07:00
parent d9fff84bd0
commit a78745189a
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
README.md
View File

@ -3,7 +3,8 @@ This repository provides the official implementation of FlashAttention from the
following paper. following paper.
**FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awareness** **FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awareness**
Tri Dao, Daniel Y. Fu, Stefano Ermon, Atri Rudra, Christopher Ré Tri Dao, Daniel Y. Fu, Stefano Ermon, Atri Rudra, Christopher Ré
Paper: https://arxiv.org/abs/2205.14135
![FlashAttention](assets/flashattn_banner.jpg) ![FlashAttention](assets/flashattn_banner.jpg)
## Alpha release (0.1). ## Alpha release (0.1).