All modules for which code is available

• lgatr.interface.axialvector
• lgatr.interface.bivector
• lgatr.interface.pseudoscalar
• lgatr.interface.scalar
• lgatr.interface.spurions
• lgatr.interface.vector
• lgatr.layers.attention.config
• lgatr.layers.attention.cross_attention
• lgatr.layers.attention.self_attention
• lgatr.layers.conditional_lgatr_block
• lgatr.layers.dropout
• lgatr.layers.layer_norm
• lgatr.layers.lgatr_block
• lgatr.layers.linear
• lgatr.layers.mlp.config
• lgatr.layers.mlp.geometric_bilinears
• lgatr.layers.mlp.mlp
• lgatr.layers.mlp.nonlinearities
• lgatr.layers.slim_layers
• lgatr.nets.conditional_lgatr
• lgatr.nets.conditional_slim
• lgatr.nets.lgatr
• lgatr.nets.slim
• lgatr.primitives.attention
• lgatr.primitives.attention_backends
• lgatr.primitives.bilinear
• lgatr.primitives.compile
• lgatr.primitives.config
• lgatr.primitives.dropout
• lgatr.primitives.invariants
• lgatr.primitives.linear
• lgatr.primitives.normalization
• lgatr.utils.autocast
• lgatr.utils.compile