GitHub repository: nshepperd/flash_attn_jax
Primary language: C++
Created: 2024-02-07T06:41:01Z
Description: JAX bindings for Flash Attention v2