jax-flash-attn3

C++ ★ 23 updated 5mo ago

JAX bindings for the flash-attention3 kernels

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