Index A | B | C | D | E | F | G | H | I | L | M | N | O | P | Q | R | S | T | V | W A AdditivePosMultiheadAttention (class in symm_learning.nn.activation) AdditiveRelMultiheadAttention (class in symm_learning.nn.activation) apply_rope() (RotaryEmbedding method) B basis_elements (GroupHomomorphismBasis attribute) basis_expansion (GroupHomomorphismBasis attribute) basis_norm_sq (GroupHomomorphismBasis attribute) bias (eConv1d property) (eConvTranspose1d property) (eLinear property) (InvariantBias property) bias_dof (InvariantBias attribute) bias_module (eAffine attribute) (eLinear attribute) broadcast_spectral_scale_and_bias() (eAffine method) C Change2DisentangledBasis (class in symm_learning.nn.disentangled) check_equivariance() (eBatchNorm1d method) (eConditionalUnet1D method) (eConv1d method) (eConvTranspose1d method) (eMultivariateNormal method) (eTimeCNNEncoder method) (eTransformerDecoderLayer method) (eTransformerEncoderLayer method) common_irreps (GroupHomomorphismBasis attribute) CommutingConstraint (class in symm_learning.nn.parametrizations) ConditionalUnet1D (class in symm_learning.models) CondTransformer (class in symm_learning.models) configure_optimizers() (CondTransformer method) cov() (in module symm_learning.stats) cov_xx (eEMAStats property) (EMAStats property) cov_xy (eEMAStats property) (EMAStats property) cov_yy (eEMAStats property) (EMAStats property) D decompose_representation() (in module symm_learning.representation_theory) dim (GroupHomomorphismBasis property) direct_sum() (in module symm_learning.representation_theory) E eAdditivePosMultiheadAttention (class in symm_learning.nn.activation) eAdditiveRelMultiheadAttention (class in symm_learning.nn.activation) eAffine (class in symm_learning.nn.linear) eBatchNorm1d (class in symm_learning.nn.normalization) eConditionalUnet1D (class in symm_learning.models) eCondTransformer (class in symm_learning.models) eConv1d (class in symm_learning.nn.conv) eConvTranspose1d (class in symm_learning.nn.conv) eEMAStats (class in symm_learning.nn.running_stats) eLayerNorm (class in symm_learning.nn.normalization) eLinear (class in symm_learning.nn.linear) EMAStats (class in symm_learning.nn.running_stats) eMLP (class in symm_learning.models) eModule (class in symm_learning.nn) eMultiheadAttention (class in symm_learning.nn.activation) eMultivariateNormal (class in symm_learning.nn.distributions) equiv_orthogonal_projection() (in module symm_learning.linalg) equiv_orthogonal_projection_coefficients() (in module symm_learning.linalg) eRMSNorm (class in symm_learning.nn.normalization) escnn_representation_form_mapping() (in module symm_learning.representation_theory) eTimeCNNEncoder (class in symm_learning.models) eTransformerDecoderLayer (class in symm_learning.nn) eTransformerEncoderLayer (class in symm_learning.nn) expand_affine() (eAffine method) expand_bias() (InvariantBias method) expand_bias_spectral_basis() (InvariantBias method) expand_weight() (eLinear method) extra_repr() (EMAStats method) F forward() (AdditivePosMultiheadAttention method) (AdditiveRelMultiheadAttention method) (Change2DisentangledBasis method) (CommutingConstraint method) (ConditionalUnet1D method) (CondTransformer method) (eAffine method) (eBatchNorm1d method) (eConditionalUnet1D method) (eCondTransformer method) (eConv1d method) (eConvTranspose1d method) (eEMAStats method) (eLayerNorm method) (EMAStats method) (eMLP method) (eMultivariateNormal method) (eRMSNorm method) (eTimeCNNEncoder method) (eTransformerDecoderLayer method) (eTransformerEncoderLayer method) (GenCondRegressor method) (GroupHomomorphismBasis method) (iMLP method) (InvariantBias method) (InvariantConstraint method) (IrrepSubspaceNormPooling method) (MLP method) (PositionalAttentionBase method) (RoPEMultiheadAttention method) (RotaryEmbedding method) (TimeCNNEncoder method) (TransformerDecoder method) (TransformerDecoderLayer method) (TransformerEncoder method) (TransformerEncoderLayer method) G G (GroupHomomorphismBasis attribute) GenCondRegressor (class in symm_learning.models) get_optim_groups() (CondTransformer method) (eCondTransformer method) GroupHomomorphismBasis (class in symm_learning.representation_theory) H has_bias (InvariantBias attribute) homo_basis (CommutingConstraint attribute) (eLinear attribute) I iMLP (class in symm_learning.models) impose_linear_equivariance() (in module symm_learning.nn.linear) in_rep (eMultivariateNormal attribute) (GroupHomomorphismBasis attribute) (InvariantBias attribute) initialize_params() (GroupHomomorphismBasis method) inv_projector (InvariantConstraint attribute) invalidate_cache() (CommutingConstraint method) (eAffine method) (eEMAStats method) (eLinear method) (EMAStats method) (eModule method) (InvariantBias method) (InvariantConstraint method) invariant_orthogonal_projector() (in module symm_learning.linalg) InvariantBias (class in symm_learning.nn.linear) InvariantConstraint (class in symm_learning.nn.parametrizations) irrep_radii() (in module symm_learning.linalg) irreps_stats() (in module symm_learning.representation_theory) IrrepSubspaceNormPooling (class in symm_learning.nn.pooling) is_complex_irreducible() (in module symm_learning.representation_theory) iso_blocks (GroupHomomorphismBasis attribute) isotypic_decomp_rep() (in module symm_learning.representation_theory) isotypic_signal2irreducible_subspaces() (in module symm_learning.linalg) L lstsq() (in module symm_learning.linalg) M mean() (in module symm_learning.stats) mean_x (eEMAStats property) (EMAStats property) mean_y (eEMAStats property) (EMAStats property) MLP (class in symm_learning.models) module symm_learning.linalg symm_learning.models symm_learning.nn symm_learning.representation_theory symm_learning.stats N n_cov_params (eMultivariateNormal attribute) num_bias_dof (eAffine property) num_scale_dof (eAffine property) O orthogonal_projection() (GroupHomomorphismBasis method) out_rep (eMultivariateNormal attribute) (GroupHomomorphismBasis attribute) (InvariantBias attribute) P permutation_matrix() (in module symm_learning.representation_theory) PositionalAttentionBase (class in symm_learning.nn.activation) project_in_isobasis() (in module symm_learning.linalg) projection_coefficients() (GroupHomomorphismBasis method) Q Q (eAffine attribute) Q_in_inv (GroupHomomorphismBasis attribute) Q_inv (eAffine attribute) Q_out (GroupHomomorphismBasis attribute) R rep (InvariantConstraint attribute) rep_x (eAffine attribute) reset_parameters() (eAffine method) (eCondTransformer method) (eConv1d method) (eConvTranspose1d method) (eLinear method) (eMLP method) (eMultiheadAttention method) (eRMSNorm method) (iMLP method) (InvariantBias method) (PositionalAttentionBase method) right_inverse() (CommutingConstraint method) (InvariantConstraint method) RoPEMultiheadAttention (class in symm_learning.nn.activation) RotaryEmbedding (class in symm_learning.nn.activation) S symm_learning.linalg module symm_learning.models module symm_learning.nn module symm_learning.representation_theory module symm_learning.stats module T tensor_cache (GroupHomomorphismBasis property) TimeCNNEncoder (class in symm_learning.models) TransformerDecoder (class in symm_learning.nn) TransformerDecoderLayer (class in symm_learning.nn) TransformerEncoder (class in symm_learning.nn) TransformerEncoderLayer (class in symm_learning.nn) V var() (in module symm_learning.stats) var_mean() (in module symm_learning.stats) W weight (eConv1d property) (eConvTranspose1d property) (eLinear property)
