kyutai-labs/flash-attn3-jax

该仓库提供了FlashAttention 3的JAX绑定，支持多种注意力机制，适用于深度学习任务的高效实现。