Index

_

__init__() (equitorch.nn.BesselBasis method)

A

add_operation_methods() (in module equitorch.structs) AddSphericalHarmonics (class in equitorch.transforms)

AddVectorNorm (class in equitorch.transforms) AlignToZWignerD (class in equitorch.nn) AnglesToMatrix (class in equitorch.nn)

B

backward_info (equitorch.nn.IrrepsLinear attribute), [1] BatchRMSNorm (class in equitorch.nn)

BesselBasis (class in equitorch.nn) both_parities() (equitorch.irreps.Irreps static method)

C

channels_divided() (equitorch.irreps.Irreps method) channels_multiplied() (equitorch.irreps.Irreps method) check_irreps() (in module equitorch.irreps) CosineCutoff (class in equitorch.nn) cpu() (equitorch.structs.IrrepsInfo method) (equitorch.structs.IrrepsLinearInfo method) (equitorch.structs.SparseProductInfo method) (equitorch.structs.SparseScaleInfo method) (equitorch.structs.TensorProductInfo method) (equitorch.structs.WignerRotationInfo method)

cuda() (equitorch.structs.IrrepsInfo method) (equitorch.structs.IrrepsLinearInfo method) (equitorch.structs.SparseProductInfo method) (equitorch.structs.SparseScaleInfo method) (equitorch.structs.TensorProductInfo method) (equitorch.structs.WignerRotationInfo method)

D

DenseWignerRotation (class in equitorch.nn) dim (equitorch.irreps.Irrep attribute) (equitorch.irreps.Irrep property) (equitorch.irreps.Irreps property)

dims() (equitorch.irreps.Irreps method) Dropout (class in equitorch.nn)

E

element_degrees() (in module equitorch.irreps) element_orders() (in module equitorch.irreps) eliminate_parity() (equitorch.irreps.Irreps method) equitorch.constants module equitorch.irreps module equitorch.nn module

equitorch.structs module equitorch.transforms module equitorch.utils module expand_left() (in module equitorch.utils) extract_batch_segments() (in module equitorch.utils) extract_scatter_indices() (in module equitorch.utils)

F

forward_info (equitorch.nn.IrrepsLinear attribute), [1]

G

Gate (class in equitorch.nn) gather_index (equitorch.structs.SparseProductInfo attribute)

generate_fully_connected_irreps_linear_paths() (in module equitorch.utils) generate_fully_connected_tp_paths() (in module equitorch.utils)

H

has_path() (in module equitorch.irreps)

I

ii0_MM0 (equitorch.structs.IrrepsLinearInfo attribute) ii0_seg_ii0MM0 (equitorch.structs.IrrepsLinearInfo attribute) index (equitorch.structs.SparseScaleInfo attribute) index1 (equitorch.structs.SparseProductInfo attribute) index2 (equitorch.structs.SparseProductInfo attribute) index_out (equitorch.structs.SparseProductInfo attribute) (equitorch.structs.SparseScaleInfo attribute) info (equitorch.nn.AlignToZWignerD attribute) (equitorch.nn.SparseWignerRotation attribute) (equitorch.nn.WignerD attribute) info_backward1 (equitorch.nn.SO2Linear attribute) info_backward2 (equitorch.nn.SO2Linear attribute) info_forward (equitorch.nn.SO2Linear attribute) info_kM1j_bwd1 (equitorch.structs.TensorProductInfo attribute) info_kM1j_bwd2 (equitorch.structs.TensorProductInfo attribute) info_kM1j_fwd (equitorch.structs.TensorProductInfo attribute) info_kM1M2_bwd1 (equitorch.structs.TensorProductInfo attribute) info_kM1M2_bwd2 (equitorch.structs.TensorProductInfo attribute) info_kM1M2_fwd (equitorch.structs.TensorProductInfo attribute) info_M_bwd1 (equitorch.structs.TensorProductInfo attribute) info_M_bwd2 (equitorch.structs.TensorProductInfo attribute) info_M_fwd (equitorch.structs.TensorProductInfo attribute) info_M_kM1M2_bwd (equitorch.structs.TensorProductInfo attribute) info_M_kM1M2_fwd (equitorch.structs.TensorProductInfo attribute) info_Mij_bwd1 (equitorch.structs.TensorProductInfo attribute) info_Mij_bwd2 (equitorch.structs.TensorProductInfo attribute) info_Mij_fwd (equitorch.structs.TensorProductInfo attribute)

initialize_linear() (in module equitorch.nn) initialize_so3_so2_linear() (in module equitorch.nn) initialize_tensor_product() (in module equitorch.nn) Irrep (class in equitorch.irreps) irrep_degrees() (in module equitorch.irreps) irrep_groups (equitorch.irreps.Irreps attribute), [1] irrep_index (equitorch.structs.IrrepsInfo attribute) irrep_indices() (in module equitorch.irreps) irrep_seg (equitorch.structs.IrrepsInfo attribute) irrep_segments() (in module equitorch.irreps) Irreps (class in equitorch.irreps) irreps_blocks_infos() (in module equitorch.utils) irreps_info (equitorch.nn.BatchRMSNorm attribute) (equitorch.nn.Gate attribute), [1] (equitorch.nn.IrrepWiseLinear attribute), [1] (equitorch.nn.LayerRMSNorm attribute) (equitorch.nn.MeanSquaredNorm attribute) (equitorch.nn.Norm attribute) (equitorch.nn.SquaredNorm attribute) (equitorch.nn.TensorDot attribute), [1] irreps_info() (in module equitorch.utils) irreps_linear_infos() (in module equitorch.utils) IrrepsInfo (class in equitorch.structs) IrrepsLinear (class in equitorch.nn) IrrepsLinearInfo (class in equitorch.structs) IrrepWiseLinear (class in equitorch.nn) isomorphic_to() (equitorch.irreps.Irreps method)

J

j_matrix() (in module equitorch.constants)

j_matrix_info (equitorch.structs.WignerRotationInfo attribute) j_matrix_info() (in module equitorch.utils)

L

l (equitorch.irreps.Irrep attribute)

l_max (equitorch.irreps.Irreps property) LayerRMSNorm (class in equitorch.nn)

M

M0_ii0MM0 (equitorch.structs.IrrepsLinearInfo attribute) M0_MM0 (equitorch.structs.IrrepsLinearInfo attribute) M_ii0MM0 (equitorch.structs.IrrepsLinearInfo attribute) M_MM0 (equitorch.structs.IrrepsLinearInfo attribute) M_out (equitorch.structs.IrrepsLinearInfo attribute) M_seg_MM0 (equitorch.structs.IrrepsLinearInfo attribute) max_channels() (equitorch.irreps.Irreps method) max_m (equitorch.structs.WignerRotationInfo attribute) MeanSquaredNorm (class in equitorch.nn)

merged() (equitorch.irreps.Irreps method) module equitorch.constants equitorch.irreps equitorch.nn equitorch.structs equitorch.transforms equitorch.utils MollifierCutoff (class in equitorch.nn)

N

Norm (class in equitorch.nn) num_gates (equitorch.nn.Gate attribute) num_irreps (equitorch.structs.IrrepsInfo attribute) num_paths (equitorch.nn.IrrepsLinear attribute) (equitorch.nn.IrrepWiseLinear attribute) (equitorch.nn.SO2Linear attribute) (equitorch.nn.SO3Linear attribute) (equitorch.nn.TensorProduct attribute)

O

out_size (equitorch.structs.IrrepsLinearInfo attribute) (equitorch.structs.SparseProductInfo attribute) (equitorch.structs.SparseScaleInfo attribute) (equitorch.structs.TensorProductInfo attribute)

P

p (equitorch.irreps.Irrep attribute) parse_irreps() (in module equitorch.irreps)

PolynomialCutoff (class in equitorch.nn) prefactor (equitorch.nn.BesselBasis attribute)

R

r_max (equitorch.nn.BesselBasis attribute) RadiusGraph (class in equitorch.transforms) rand_rotation_angles() (in module equitorch.utils) rand_rotation_matrices() (in module equitorch.utils) rand_spherical_angles() (in module equitorch.utils) rand_spherical_xyz() (in module equitorch.utils) rdims (equitorch.structs.IrrepsInfo attribute)

reset_parameters() (equitorch.nn.BatchRMSNorm method) (equitorch.nn.LayerRMSNorm method) reset_running_stats() (equitorch.nn.BatchRMSNorm method) rotate_z_info_bwd_cs (equitorch.structs.WignerRotationInfo attribute) rotate_z_info_bwd_input (equitorch.structs.WignerRotationInfo attribute) rotate_z_info_fwd (equitorch.structs.WignerRotationInfo attribute) rsqrt_dims (equitorch.structs.IrrepsInfo attribute)

S

scale (equitorch.structs.SparseProductInfo attribute) (equitorch.structs.SparseScaleInfo attribute) scale_MM0 (equitorch.structs.IrrepsLinearInfo attribute) scales_ii0 (equitorch.structs.IrrepsLinearInfo attribute) seg_out (equitorch.structs.SparseProductInfo attribute) (equitorch.structs.SparseScaleInfo attribute) Separable (class in equitorch.nn) short_repr() (equitorch.irreps.Irreps method) show_irreps() (in module equitorch.irreps) sign (equitorch.structs.WignerRotationInfo attribute) simplified() (equitorch.irreps.Irreps method) SinCos (class in equitorch.nn) so2_linear_info() (in module equitorch.utils) so2_linear_infos() (in module equitorch.utils) SO2Linear (class in equitorch.nn) so3_clebsch_gordan() (in module equitorch.constants)

SO3Linear (class in equitorch.nn) sort_by_column_key() (in module equitorch.utils) sorted() (equitorch.irreps.Irreps method) sparse_product_info() (in module equitorch.utils) sparse_product_infos() (in module equitorch.utils) sparse_scale_info() (in module equitorch.utils) sparse_scale_infos() (in module equitorch.utils) SparseProductInfo (class in equitorch.structs) SparseScaleInfo (class in equitorch.structs) SparseWignerRotation (class in equitorch.nn) spherical_harmonics() (equitorch.irreps.Irreps static method) SphericalHarmonics (class in equitorch.nn) SphericalToXYZ (class in equitorch.nn) SplitIrreps (class in equitorch.nn) SquaredNorm (class in equitorch.nn) squeeze_max_channels() (equitorch.irreps.Irreps method)

T

TensorDot (class in equitorch.nn) TensorProduct (class in equitorch.nn) TensorProductInfo (class in equitorch.structs) to() (equitorch.structs.IrrepsInfo method) (equitorch.structs.IrrepsLinearInfo method) (equitorch.structs.SparseProductInfo method) (equitorch.structs.SparseScaleInfo method) (equitorch.structs.TensorProductInfo method) (equitorch.structs.WignerRotationInfo method)

tp_info() (in module equitorch.utils) tp_info_backward1 (equitorch.nn.SO3Linear attribute), [1] (equitorch.nn.TensorProduct attribute), [1] tp_info_backward2 (equitorch.nn.SO3Linear attribute), [1] (equitorch.nn.TensorProduct attribute), [1] tp_info_forward (equitorch.nn.SO3Linear attribute), [1] (equitorch.nn.TensorProduct attribute), [1] tp_infos() (in module equitorch.utils)

U

unique_irreps() (in module equitorch.irreps)

W

weight (equitorch.nn.IrrepsLinear attribute) (equitorch.nn.IrrepWiseLinear attribute) (equitorch.nn.SO2Linear attribute) (equitorch.nn.SO3Linear attribute) (equitorch.nn.TensorProduct attribute) weight_numel (equitorch.nn.IrrepsLinear attribute) (equitorch.nn.IrrepWiseLinear attribute) (equitorch.nn.SO2Linear attribute) (equitorch.nn.SO3Linear attribute) (equitorch.nn.TensorProduct attribute)

wigner_d_info() (in module equitorch.utils) WignerD (class in equitorch.nn) WignerRotationInfo (class in equitorch.structs)

X

XYZToSinCos (class in equitorch.nn)

XYZToSpherical (class in equitorch.nn)

Z

z_rotation_infos() (in module equitorch.utils)