alternative flash-attention

[Reidmen]

Reference, one-shot implementation:

https://github.com/nebius/kvax