Name		Name	Last commit message	Last commit date
README.md		README.md
fa2.py		fa2.py
pyproject.toml		pyproject.toml
uv.lock		uv.lock

Repository files navigation

FlashAttention2

fa2.py contains 3 different attention implementations:

Vanilla attention in PyTorch
FlashAttention2 in PyTorch
FlashAttention2 in Triton

About

FlashAttention2 in Triton and PyTorch

0 stars

0 watching

Report repository

Releases

Packages

Contributors

Languages