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: 2
Downloads last week: 36
Downloads last month: 225