Comment by platers

Flash attention natively supports packing multiple variable length sequences into a single call, what is the advantage of jagged flash attention?