Undergrad@University of Science and Technology of China.
dune-z / flash-attention Goto Github PK
View Code? Open in Web Editor NEWThis project forked from dao-ailab/flash-attention
Fast and memory-efficient exact attention
License: BSD 3-Clause "New" or "Revised" License