flash-attn3-jax

C++ ★ 26 updated 3mo ago

JAX bindings for the FlashAttention 3 kernels

No plain-English explanation yet — one is being written right now. Check back in a minute.