PyRank

Flash Attention 2 Python Packages

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

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

271 26 0
erfanzar
jax-flash-attn2

Flash Attention Implementation with Multiple Backend Support and Sharding This module provides a flexible implementation of Flash Attention with support for different backends (GPU, TPU, CPU) and platforms (Triton, Pallas, JAX).

166 34 1