Latest version: v0.0.1
The information on this page was curated by experts in our Cybersecurity Intelligence Team.
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).
No known vulnerabilities found