flash-attn-triton

PyPI page
Home page
Author: Alyssa Vance
Summary: Triton-based backend for Flash Attention 2
Latest version: 0.1.1
Required dependencies: einops | packaging | pytest | torch | triton

Downloads last day: 8
Downloads last week: 64
Downloads last month: 313