Linear Attention https://github.com/fla-org/flash-linear-attention flash-linear-attention transformers==4.45.0 Ref