I don't understand the long part, as we already have zigzag examples

https://github.com/zhuzilin/ring-flash-attention