/ring-flash-attention

Ring attention implementation with flash attention

Primary LanguagePython

Stargazers