nshepperd / flash_attn_jaxView on GitHub
JAX bindings for Flash Attention v2
104Feb 28, 2026Updated last month

Alternatives and similar repositories for flash_attn_jax

Users that are interested in flash_attn_jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.

Sorting:

Are these results useful?