API Reference

Modules:

• equitorch.constants
  • j_matrix()
  • so3_clebsch_gordan()
• equitorch.irreps
  • Irrep
  • parse_irreps()
  • show_irreps()
  • check_irreps()
  • has_path()
  • Irreps
  • unique_irreps()
  • irrep_segments()
  • irrep_indices()
  • irrep_degrees()
  • element_degrees()
  • element_orders()
• equitorch.nn
  • SO3Linear
  • IrrepWiseLinear
  • IrrepsLinear
  • SO2Linear
  • SplitIrreps
  • Separable
  • SphericalHarmonics
  • XYZToSpherical
  • SphericalToXYZ
  • XYZToSinCos
  • BatchRMSNorm
  • LayerRMSNorm
  • initialize_tensor_product()
  • initialize_so3_so2_linear()
  • initialize_linear()
  • TensorProduct
  • TensorDot
  • SparseWignerRotation
  • DenseWignerRotation
  • WignerD
  • AlignToZWignerD
  • PolynomialCutoff
  • CosineCutoff
  • MollifierCutoff
  • Gate
  • SinCos
  • BesselBasis
  • SquaredNorm
  • Norm
  • MeanSquaredNorm
  • Dropout
  • AnglesToMatrix
• equitorch.structs
  • add_operation_methods()
  • SparseScaleInfo
  • SparseProductInfo
  • TensorProductInfo
  • IrrepsInfo
  • IrrepsLinearInfo
  • WignerRotationInfo
• equitorch.transforms
  • RadiusGraph
  • AddSphericalHarmonics
  • AddVectorNorm
• equitorch.utils
  • rand_spherical_xyz()
  • rand_spherical_angles()
  • rand_rotation_angles()
  • rand_rotation_matrices()
  • expand_left()
  • extract_batch_segments()
  • sort_by_column_key()
  • extract_scatter_indices()
  • sparse_scale_info()
  • sparse_scale_infos()
  • sparse_product_info()
  • sparse_product_infos()
  • generate_fully_connected_tp_paths()
  • tp_info()
  • tp_infos()
  • generate_fully_connected_irreps_linear_paths()
  • irreps_linear_infos()
  • irreps_info()
  • z_rotation_infos()
  • j_matrix_info()
  • wigner_d_info()
  • irreps_blocks_infos()
  • so2_linear_info()
  • so2_linear_infos()