Kvax: Fast and easy-to-use flash attention implementation for Jax #1 Post by exizt88 » Sun, Mar 02, 2025, 10:30 PM UTC Kvax: Fast and easy-to-use flash attention implementation for Jaxgithub.com