nshepperd / flash_attn_jaxLinks

JAX bindings for Flash Attention v2
89Updated 11 months ago

Alternatives and similar repositories for flash_attn_jax

Users that are interested in flash_attn_jax are comparing it to the libraries listed below

Sorting: