lucidrains / flash-attention-jaxLinks

Implementation of Flash Attention in Jax
212Updated last year

Alternatives and similar repositories for flash-attention-jax

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

Sorting: