Index A | B | C | D | E | F | G | H | I | J | L | M | N | O | P | R | S | T | U | V | X | Y A A (cuequivariance.Rep property) add_or_rename_modes() (cuequivariance.SegmentedTensorProduct method) add_or_transpose_modes() (cuequivariance.SegmentedTensorProduct method) add_path() (cuequivariance.SegmentedTensorProduct method) add_segment() (cuequivariance.SegmentedOperand method) (cuequivariance.SegmentedTensorProduct method) add_segments() (cuequivariance.SegmentedTensorProduct method) algebra() (cuequivariance.IrrepsAndLayout method) (cuequivariance.O3 method) (cuequivariance.Rep method) (cuequivariance.SO3 method) (cuequivariance.SU2 method) all_same_segment_shape() (cuequivariance.EquivariantPolynomial method) (cuequivariance.SegmentedOperand method) (cuequivariance.SegmentedPolynomial method) (cuequivariance.SegmentedTensorProduct method) all_segments_are_used() (cuequivariance.SegmentedTensorProduct method) append_modes_to_all_operands() (cuequivariance.SegmentedTensorProduct method) apply_fn() (cuequivariance.EquivariantPolynomial method) (cuequivariance.SegmentedPolynomial method) as_irreps_array() (in module cuequivariance_jax) assert_valid() (cuequivariance.SegmentedOperand method) assume (class in cuequivariance) B backward() (cuequivariance.EquivariantPolynomial method) (cuequivariance.SegmentedPolynomial method) BatchNorm (class in cuequivariance_torch.layers) C canonicalize_subscripts() (cuequivariance.EquivariantPolynomial method) (cuequivariance.SegmentedPolynomial method) (cuequivariance.SegmentedTensorProduct method) change_layout() (cuequivariance_jax.RepArray method) channelwise_tensor_product() (in module cuequivariance.group_theory.descriptors) ChannelWiseTensorProduct (class in cuequivariance_torch) clebsch_gordan() (cuequivariance.Irrep class method) (cuequivariance.O3 class method) (cuequivariance.SO3 class method) (cuequivariance.SU2 class method) (in module cuequivariance) coefficients_are_stackable (cuequivariance.SegmentedTensorProduct property) coefficients_equal_one() (cuequivariance.SegmentedTensorProduct method) compressed_path_segment() (cuequivariance.SegmentedTensorProduct method) compute_only() (cuequivariance.EquivariantPolynomial method) (cuequivariance.SegmentedPolynomial method) concatenate() (cuequivariance.SegmentedPolynomial class method) (in module cuequivariance_jax) consolidate() (cuequivariance.EquivariantPolynomial method) (cuequivariance.SegmentedPolynomial method) consolidate_modes() (cuequivariance.SegmentedTensorProduct method) consolidate_paths() (cuequivariance.SegmentedTensorProduct method) continuous_generators() (cuequivariance.IrrepsAndLayout method) (cuequivariance.O3 method) (cuequivariance.Rep method) (cuequivariance.SO3 method) (cuequivariance.SU2 method) copy() (cuequivariance.SegmentedOperand method) count() (cuequivariance.Irreps method) cuequivariance module cuequivariance.group_theory.descriptors module cuequivariance_jax module cuequivariance_torch module D dim (cuequivariance.Irreps property) (cuequivariance.O3 property) (cuequivariance.Rep property) (cuequivariance.SO3 property) (cuequivariance.SU2 property) discrete_generators() (cuequivariance.IrrepsAndLayout method) (cuequivariance.O3 method) (cuequivariance.Rep method) (cuequivariance.SO3 method) (cuequivariance.SU2 method) dtype (cuequivariance_jax.RepArray property) E elementwise_tensor_product() (in module cuequivariance.group_theory.descriptors) empty_segments() (cuequivariance.SegmentedOperand class method) (cuequivariance.SegmentedTensorProduct class method) encode_rotation_angle() (in module cuequivariance_torch) equivariant_polynomial() (in module cuequivariance_jax) EquivariantPolynomial (class in cuequivariance) EquivariantTensorProduct (class in cuequivariance_torch) eval_last_operand() (cuequivariance.SegmentedPolynomial class method) exp_map() (cuequivariance.Rep method) F filter() (cuequivariance.Irreps method) (cuequivariance_jax.RepArray method) filter_drop_unsued_operands() (cuequivariance.EquivariantPolynomial method) (cuequivariance.SegmentedPolynomial method) filter_keep_operands() (cuequivariance.EquivariantPolynomial method) (cuequivariance.SegmentedPolynomial method) filter_keep_outputs() (cuequivariance.EquivariantPolynomial method) (cuequivariance.SegmentedPolynomial method) flatten_coefficient_modes() (cuequivariance.EquivariantPolynomial method) (cuequivariance.SegmentedPolynomial method) (cuequivariance.SegmentedTensorProduct method) flatten_modes() (cuequivariance.EquivariantPolynomial method) (cuequivariance.SegmentedPolynomial method) (cuequivariance.SegmentedTensorProduct method) flop() (cuequivariance.EquivariantPolynomial method) (cuequivariance.SegmentedPolynomial method) (cuequivariance.SegmentedTensorProduct method) forward() (cuequivariance_torch.ChannelWiseTensorProduct method) (cuequivariance_torch.EquivariantTensorProduct method) (cuequivariance_torch.FullyConnectedTensorProduct method) (cuequivariance_torch.Inversion method) (cuequivariance_torch.layers.BatchNorm method) (cuequivariance_torch.layers.FullyConnectedTensorProductConv method) (cuequivariance_torch.Linear method) (cuequivariance_torch.Rotation method) (cuequivariance_torch.SegmentedPolynomial method) (cuequivariance_torch.SphericalHarmonics method) (cuequivariance_torch.SymmetricContraction method) (cuequivariance_torch.TensorProduct method) (cuequivariance_torch.TransposeIrrepsLayout method) from_base64() (cuequivariance.SegmentedTensorProduct class method) from_bytes() (cuequivariance.SegmentedTensorProduct class method) from_dict() (cuequivariance.SegmentedTensorProduct class method) from_json() (cuequivariance.SegmentedTensorProduct class method) from_segments() (in module cuequivariance_jax) from_string() (cuequivariance.Irrep class method) (cuequivariance.O3 class method) (cuequivariance.SO3 class method) (cuequivariance.SU2 class method) from_subscripts() (cuequivariance.SegmentedTensorProduct class method) fully_connected_tensor_product() (in module cuequivariance.group_theory.descriptors) FullyConnectedTensorProduct (class in cuequivariance_torch) FullyConnectedTensorProductConv (class in cuequivariance_torch.layers) fuse_paths_with_same_indices() (cuequivariance.SegmentedTensorProduct method) fuse_stps() (cuequivariance.EquivariantPolynomial method) (cuequivariance.SegmentedPolynomial method) G get_dimensions_dict() (cuequivariance.SegmentedOperand method) (cuequivariance.SegmentedTensorProduct method) get_dims() (cuequivariance.SegmentedOperand method) (cuequivariance.SegmentedTensorProduct method) get_path_dim() (cuequivariance.SegmentedTensorProduct method) get_path_dimensions_dict() (cuequivariance.SegmentedTensorProduct method) get_segment_shape() (cuequivariance.SegmentedTensorProduct method) group_by_idential_buffers() (cuequivariance.Operation static method) group_by_operational_symmetries() (cuequivariance.Operation static method) H H (cuequivariance.Rep property) I indices (cuequivariance.SegmentedTensorProduct property) insert_path() (cuequivariance.SegmentedTensorProduct method) insert_segment() (cuequivariance.SegmentedOperand method) insert_segments() (cuequivariance.SegmentedTensorProduct method) Inversion (class in cuequivariance_torch) ir_mul (cuequivariance.IrrepsLayout attribute) Irrep (class in cuequivariance) Irreps (class in cuequivariance) irreps (cuequivariance_jax.RepArray property) IrrepsAndLayout (class in cuequivariance) IrrepsLayout (class in cuequivariance) is_irreps_array() (cuequivariance_jax.RepArray method) is_scalar() (cuequivariance.Irreps method) (cuequivariance.IrrepsAndLayout method) (cuequivariance.O3 method) (cuequivariance.Rep method) (cuequivariance.SU2 method) is_trivial() (cuequivariance.Rep method) iterator() (cuequivariance.Irrep class method) (cuequivariance.O3 class method) (cuequivariance.SO3 class method) (cuequivariance.SU2 class method) J jvp() (cuequivariance.EquivariantPolynomial method) (cuequivariance.Operation method) (cuequivariance.SegmentedPolynomial method) L LayerNorm (class in cuequivariance_jax.flax_linen) layout (cuequivariance_jax.RepArray property) layout_insensitive() (cuequivariance.Irreps method) lie_dim (cuequivariance.Rep property) Linear (class in cuequivariance_torch) linear() (in module cuequivariance.group_theory.descriptors) M memory() (cuequivariance.EquivariantPolynomial method) (cuequivariance.SegmentedPolynomial method) (cuequivariance.SegmentedTensorProduct method) merge_consecutive() (cuequivariance.Irreps method) modify_coefficients() (cuequivariance.SegmentedTensorProduct method) module cuequivariance cuequivariance.group_theory.descriptors cuequivariance_jax cuequivariance_torch move_axis_to_mul() (cuequivariance_jax.RepArray method) move_operand() (cuequivariance.SegmentedTensorProduct method) move_operand_first() (cuequivariance.SegmentedTensorProduct method) move_operand_last() (cuequivariance.SegmentedTensorProduct method) mul_ir (cuequivariance.IrrepsLayout attribute) muls (cuequivariance.Irreps property) N ndim (cuequivariance_jax.RepArray property) new_scalars() (cuequivariance.Irreps method) normalize_paths_for_operand() (cuequivariance.SegmentedTensorProduct method) num_inputs (cuequivariance.EquivariantPolynomial property) (cuequivariance.SegmentedPolynomial property) num_irreps (cuequivariance.Irreps property) num_operands (cuequivariance.EquivariantPolynomial property) (cuequivariance.SegmentedPolynomial property) (cuequivariance.SegmentedTensorProduct property) num_outputs (cuequivariance.EquivariantPolynomial property) (cuequivariance.SegmentedPolynomial property) num_paths (cuequivariance.SegmentedTensorProduct property) num_segments (cuequivariance.SegmentedOperand property) O O3 (class in cuequivariance) operands (cuequivariance.EquivariantPolynomial property) (cuequivariance.SegmentedPolynomial property) operands_with_identical_buffers() (cuequivariance.Operation method) operands_with_identical_segments() (cuequivariance.SegmentedTensorProduct method) Operation (class in cuequivariance) P permute_operands() (cuequivariance.SegmentedTensorProduct method) permute_segments() (cuequivariance.SegmentedTensorProduct method) R randn() (in module cuequivariance_jax) regexp_pattern() (cuequivariance.Irrep class method) (cuequivariance.O3 class method) (cuequivariance.SO3 class method) (cuequivariance.SU2 class method) regroup() (cuequivariance.Irreps method) (cuequivariance_jax.RepArray method) remove_empty_segments() (cuequivariance.SegmentedTensorProduct method) remove_zero_multiplicities() (cuequivariance.Irreps method) remove_zero_paths() (cuequivariance.SegmentedTensorProduct method) Rep (class in cuequivariance) rep() (cuequivariance_jax.RepArray method) RepArray (class in cuequivariance_jax) Rotation (class in cuequivariance_torch) rotation() (cuequivariance.SO3 method) round_coefficients_to_rational() (cuequivariance.SegmentedTensorProduct method) round_coefficients_to_sqrt_rational() (cuequivariance.SegmentedTensorProduct method) S segment_shape (cuequivariance.SegmentedOperand property) segment_size (cuequivariance.SegmentedOperand property) segment_slice() (cuequivariance.SegmentedTensorProduct method) segment_slices() (cuequivariance.SegmentedOperand method) segmented_polynomial() (in module cuequivariance_jax) SegmentedOperand (class in cuequivariance) SegmentedPolynomial (class in cuequivariance) (class in cuequivariance_torch) SegmentedTensorProduct (class in cuequivariance) segments (cuequivariance_jax.RepArray property) set_mul() (cuequivariance.Irreps method) shape (cuequivariance_jax.RepArray property) shape() (cuequivariance.IrrepsLayout method) simplify() (cuequivariance.Irreps method) size (cuequivariance.SegmentedOperand property) slice_by_mul (cuequivariance_jax.RepArray property) slices() (cuequivariance.Irreps method) SO3 (class in cuequivariance) sort() (cuequivariance.Irreps method) (cuequivariance_jax.RepArray method) sort_indices_for_identical_operands() (cuequivariance.SegmentedTensorProduct method) sort_paths() (cuequivariance.SegmentedTensorProduct method) spherical_harmonics() (in module cuequivariance.group_theory.descriptors) (in module cuequivariance_jax) SphericalHarmonics (class in cuequivariance_torch) split_mode() (cuequivariance.SegmentedTensorProduct method) squeeze_modes() (cuequivariance.EquivariantPolynomial method) (cuequivariance.SegmentedPolynomial method) (cuequivariance.SegmentedTensorProduct method) stack() (cuequivariance.EquivariantPolynomial class method) (cuequivariance.SegmentedOperand class method) (cuequivariance.SegmentedPolynomial class method) stack_tensor_products() (cuequivariance.SegmentedPolynomial class method) stacked_coefficients (cuequivariance.SegmentedTensorProduct property) SU2 (class in cuequivariance) subscripts (cuequivariance.SegmentedTensorProduct property) symmetric_contraction() (in module cuequivariance.group_theory.descriptors) SymmetricContraction (class in cuequivariance_torch) symmetries() (cuequivariance.SegmentedTensorProduct method) symmetrize_for_identical_operands() (cuequivariance.EquivariantPolynomial method) (cuequivariance.SegmentedPolynomial method) symmetrize_operands() (cuequivariance.SegmentedTensorProduct method) T TensorProduct (class in cuequivariance_torch) to_base64() (cuequivariance.SegmentedTensorProduct method) to_bytes() (cuequivariance.SegmentedTensorProduct method) to_dict() (cuequivariance.SegmentedTensorProduct method) to_json() (cuequivariance.SegmentedTensorProduct method) to_text() (cuequivariance.SegmentedTensorProduct method) transform() (cuequivariance_jax.RepArray method) transpose() (cuequivariance.EquivariantPolynomial method) (cuequivariance.Operation method) (cuequivariance.SegmentedPolynomial method) TransposeIrrepsLayout (class in cuequivariance_torch) trivial() (cuequivariance.Irrep class method) (cuequivariance.IrrepsAndLayout method) (cuequivariance.Rep method) U unsymmetrize_for_identical_operands() (cuequivariance.EquivariantPolynomial method) (cuequivariance.SegmentedPolynomial method) used_inputs() (cuequivariance.EquivariantPolynomial method) (cuequivariance.SegmentedPolynomial method) used_operands() (cuequivariance.EquivariantPolynomial method) (cuequivariance.SegmentedPolynomial method) used_outputs() (cuequivariance.EquivariantPolynomial method) (cuequivariance.SegmentedPolynomial method) V vector_to_euler_angles() (in module cuequivariance_torch) X X (cuequivariance.Rep property) x_rotation() (in module cuequivariance.group_theory.descriptors) Y y_rotation() (in module cuequivariance.group_theory.descriptors) yxy_rotation() (in module cuequivariance.group_theory.descriptors)
