PyRank

Flash Attention 2 Python Packages

Python packages with the GitHub topic flash-attention-2. Sorted by monthly downloads, with stars and forks.
egaoharu-kensei
flash-attention-triton

Cross-platform FlashAttention-2 Triton implementation for Turing+ GPUs with custom configuration mode

114 33 0
erfanzar
jax-flash-attn2

A flexible and efficient implementation of Flash Attention 2.0 for JAX, supporting multiple backends (GPU/TPU/CPU) and platforms (Triton/Pallas/JAX).

81 34 1