GitHub repository: lucidrains/flash-attention-jax
Primary language: Python
Created: 2022-07-12T03:41:20Z
Description: Implementation of Flash Attention in Jax