GitHub - tspeterkim/flash-attention-minimal: Flash Attention in ~100 lines of CUDA (forward pass only) - discu.eu

Hacker News

Show HN: Flash Attention in ~100 lines of CUDA https://github.com/tspeterkim/flash-attention-minimal 40 comments 16/3/2024

Reddit

[P] Flash Attention in ~100 lines of CUDA https://github.com/tspeterkim/flash-attention-minimal 2 comments 8/3/2024 machinelearning

Linked pages

[2205.14135] FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awareness https://arxiv.org/abs/2205.14135 3 comments

Would you like to stay up to date with Computer science? Checkout Computer science Weekly.

Related searches:

Search whole site: site:github.com

Search title: GitHub - tspeterkim/flash-attention-minimal: Flash Attention in ~100 lines of CUDA (forward pass only)

See how to search.

Submit link to: