Name		Name	Last commit message	Last commit date
Latest commit History 9 Commits
attention		attention
tests		tests
.gitignore		.gitignore
README.md		README.md
poetry.lock		poetry.lock
pyproject.toml		pyproject.toml

Repository files navigation

Flash Attention

Toy example implementation of FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awareness

About

Toy Flash Attention implementation in torch

torch flash-attention flash-attention-2 flash-attention-3

Report repository

Releases

No releases published

Packages

Contributors

Languages

Python 100.0%