kyutai-labs / jax-flash-attn3

JAX bindings for the flash-attention3 kernels
11Updated 7 months ago

Alternatives and similar repositories for jax-flash-attn3:

Users that are interested in jax-flash-attn3 are comparing it to the libraries listed below