HomeAI Tutorial

jax-flash-attn2

Public

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

निर्माण समय2024-10-24T02:08:50
अपडेट समय2025-03-05T02:57:36
31
Stars
1
Stars Increase

संबंधित प्रोजेक्ट