GitHub repository: kyutai-labs/jax-flash-attn3
Primary language: C++
Description: JAX bindings for the flash-attention3 kernels