Toy Flash Attention implementation in torch
-
Updated
Sep 22, 2024 - Python
Toy Flash Attention implementation in torch
Add a description, image, and links to the flash-attention-3 topic page so that developers can more easily learn about it.
To associate your repository with the flash-attention-3 topic, visit your repo's landing page and select "manage topics."