Triton implementation

[hp-l33]

Thank you for your excellent work!

I maintain a library implementing bi-directional linear attention with Triton, which now now supports some of your Flatten-Transformer models. By the way, it also provides preliminary support for MLLA.

Best wishes!

[tian-qing001]

Hi @hp-l33, thanks for your efforts. I believe the fast implementation will be beneficial for the community~