return flash attention#273

JamesScott-swe wants to merge 1 commit intomainevolutionaryscale/esm:mainfrom

james_scott/return_flash_attentionevolutionaryscale/esm:james_scott/return_flash_attentionCopy head branch name to clipboard