Support efficient flash attention for packed sequences using flash-attn-2.0 - #350

Support efficient flash attention for packed sequences using flash-attn-2.0#350

lucidrains merged 3 commits into

lucidrains:mainlucidrains/x-transformers:mainfrom

muthissar:flash-pack-seqmuthissar/x-transformers:flash-pack-seqCopy head branch name to clipboard