[lgatr.primitives.attention_backends]
default_attention = lgatr.primitives.attention_backends.default_attention
flex_attention = lgatr.primitives.attention_backends.flex_attention
xformers_attention = lgatr.primitives.attention_backends.xformers_attention
