/flash-attention-ipu

Poplar implementation of FlashAttention for IPU

Primary LanguageC++MIT LicenseMIT

Watchers