Index

A

abs_squared_norm() (in module lgatr.primitives.invariants) activation (lgatr.layers.mlp.config.MLPConfig attribute) additional_k_mv_channels (lgatr.layers.attention.config.CrossAttentionConfig attribute) additional_k_s_channels (lgatr.layers.attention.config.CrossAttentionConfig attribute)

additional_q_mv_channels (lgatr.layers.attention.config.CrossAttentionConfig attribute) additional_q_s_channels (lgatr.layers.attention.config.CrossAttentionConfig attribute) additional_qk_mv_channels (lgatr.layers.attention.config.SelfAttentionConfig attribute) additional_qk_s_channels (lgatr.layers.attention.config.SelfAttentionConfig attribute)

C

cast() (lgatr.layers.attention.config.CrossAttentionConfig class method) (lgatr.layers.attention.config.SelfAttentionConfig class method) (lgatr.layers.mlp.config.MLPConfig class method)

ConditionalLGATr (class in lgatr.nets.conditional_lgatr) ConditionalLGATrBlock (class in lgatr.layers.conditional_lgatr_block) CrossAttention (class in lgatr.layers.attention.cross_attention) CrossAttentionConfig (class in lgatr.layers.attention.config)

D

dropout_prob (lgatr.layers.attention.config.CrossAttentionConfig attribute) (lgatr.layers.attention.config.SelfAttentionConfig attribute) (lgatr.layers.mlp.config.MLPConfig attribute)

E

embed_axialvector() (in module lgatr.interface.axialvector) embed_pseudoscalar() (in module lgatr.interface.pseudoscalar) embed_scalar() (in module lgatr.interface.scalar) embed_vector() (in module lgatr.interface.vector) equi_layer_norm() (in module lgatr.primitives.normalization) equi_linear() (in module lgatr.primitives.linear)

EquiLayerNorm (class in lgatr.layers.layer_norm) EquiLinear (class in lgatr.layers.linear) extract_axialvector() (in module lgatr.interface.axialvector) extract_pseudoscalar() (in module lgatr.interface.pseudoscalar) extract_scalar() (in module lgatr.interface.scalar) extract_vector() (in module lgatr.interface.vector)

F

forward() (lgatr.layers.attention.cross_attention.CrossAttention method) (lgatr.layers.attention.self_attention.SelfAttention method) (lgatr.layers.conditional_lgatr_block.ConditionalLGATrBlock method) (lgatr.layers.dropout.GradeDropout method) (lgatr.layers.layer_norm.EquiLayerNorm method) (lgatr.layers.lgatr_block.LGATrBlock method) (lgatr.layers.linear.EquiLinear method) (lgatr.layers.mlp.geometric_bilinears.GeometricBilinear method) (lgatr.layers.mlp.mlp.GeoMLP method) (lgatr.layers.mlp.nonlinearities.ScalarGatedNonlinearity method) (lgatr.nets.conditional_lgatr.ConditionalLGATr method) (lgatr.nets.lgatr.LGATr method)

G

gated_gelu() (in module lgatr.primitives.nonlinearities) gated_relu() (in module lgatr.primitives.nonlinearities) gated_sigmoid() (in module lgatr.primitives.nonlinearities) geometric_product() (in module lgatr.primitives.bilinear) GeometricBilinear (class in lgatr.layers.mlp.geometric_bilinears) GeoMLP (class in lgatr.layers.mlp.mlp)

get_num_spurions() (in module lgatr.interface.spurions) get_spurions() (in module lgatr.interface.spurions) grade_dropout() (in module lgatr.primitives.dropout) grade_involute() (in module lgatr.primitives.linear) grade_project() (in module lgatr.primitives.linear) GradeDropout (class in lgatr.layers.dropout)

H

head_scale (lgatr.layers.attention.config.CrossAttentionConfig attribute) (lgatr.layers.attention.config.SelfAttentionConfig attribute) hidden_mv_channels (lgatr.layers.attention.config.CrossAttentionConfig property) (lgatr.layers.attention.config.SelfAttentionConfig property)

hidden_s_channels (lgatr.layers.attention.config.CrossAttentionConfig property) (lgatr.layers.attention.config.SelfAttentionConfig property)

I

in_kv_mv_channels (lgatr.layers.attention.config.CrossAttentionConfig attribute) in_kv_s_channels (lgatr.layers.attention.config.CrossAttentionConfig attribute) in_mv_channels (lgatr.layers.attention.config.SelfAttentionConfig attribute) in_q_mv_channels (lgatr.layers.attention.config.CrossAttentionConfig attribute) in_q_s_channels (lgatr.layers.attention.config.CrossAttentionConfig attribute)

in_s_channels (lgatr.layers.attention.config.SelfAttentionConfig attribute) increase_hidden_channels (lgatr.layers.attention.config.CrossAttentionConfig attribute) (lgatr.layers.attention.config.SelfAttentionConfig attribute) (lgatr.layers.mlp.config.MLPConfig attribute) inner_product() (in module lgatr.primitives.invariants)

L

LGATr (class in lgatr.nets.lgatr) lgatr.interface.axialvector module lgatr.interface.pseudoscalar module lgatr.interface.scalar module lgatr.interface.spurions module lgatr.interface.vector module lgatr.primitives.attention module

lgatr.primitives.bilinear module lgatr.primitives.dropout module lgatr.primitives.invariants module lgatr.primitives.linear module lgatr.primitives.nonlinearities module lgatr.primitives.normalization module LGATrBlock (class in lgatr.layers.lgatr_block) LGATrConfig (class in lgatr.primitives.config)

M

mix_pseudoscalar_into_scalar (lgatr.primitives.config.LGATrConfig attribute) MLPConfig (class in lgatr.layers.mlp.config) module lgatr.interface.axialvector lgatr.interface.pseudoscalar lgatr.interface.scalar lgatr.interface.spurions lgatr.interface.vector lgatr.primitives.attention lgatr.primitives.bilinear lgatr.primitives.dropout lgatr.primitives.invariants lgatr.primitives.linear lgatr.primitives.nonlinearities lgatr.primitives.normalization

multi_query (lgatr.layers.attention.config.CrossAttentionConfig attribute) (lgatr.layers.attention.config.SelfAttentionConfig attribute) mv_channels (lgatr.layers.mlp.config.MLPConfig attribute)

N

num_heads (lgatr.layers.attention.config.CrossAttentionConfig attribute) (lgatr.layers.attention.config.SelfAttentionConfig attribute)

num_hidden_layers (lgatr.layers.mlp.config.MLPConfig attribute) num_pin_linear_basis_elements (lgatr.primitives.config.LGATrConfig property)

O

out_mv_channels (lgatr.layers.attention.config.CrossAttentionConfig attribute) (lgatr.layers.attention.config.SelfAttentionConfig attribute) out_s_channels (lgatr.layers.attention.config.CrossAttentionConfig attribute) (lgatr.layers.attention.config.SelfAttentionConfig attribute)

output_init (lgatr.layers.attention.config.CrossAttentionConfig attribute) (lgatr.layers.attention.config.SelfAttentionConfig attribute)

R

reset_parameters() (lgatr.layers.linear.EquiLinear method)

reverse() (in module lgatr.primitives.linear)

S

s_channels (lgatr.layers.mlp.config.MLPConfig attribute) ScalarGatedNonlinearity (class in lgatr.layers.mlp.nonlinearities) scaled_dot_product_attention() (in module lgatr.primitives.attention)

sdp_attention() (in module lgatr.primitives.attention) SelfAttention (class in lgatr.layers.attention.self_attention) SelfAttentionConfig (class in lgatr.layers.attention.config)

U

use_bivector (lgatr.primitives.config.LGATrConfig attribute)

use_fully_connected_subgroup (lgatr.primitives.config.LGATrConfig attribute) use_geometric_product (lgatr.primitives.config.LGATrConfig attribute)