API

• Irreps
  • Irrep
  • Irreps
• IrrepsArray
  • IrrepsArray
  • from_chunks()
  • as_irreps_array()
  • zeros()
  • zeros_like()
  • concatenate()
  • mean()
  • norm()
  • dot()
  • cross()
  • normal()
  • sum()
  • where()
• Tensor Products
  • spherical_harmonics()
  • tensor_product()
  • tensor_square()
  • reduced_tensor_product_basis()
  • reduced_symmetric_tensor_product_basis()
  • reduced_antisymmetric_tensor_product_basis()
  • elementwise_tensor_product()
  • tensor_product_with_spherical_harmonics()
  • sh()
• Equivariant Operations
  • grad()
• Basic functions for Rotations
  • rand_matrix()
  • identity_angles()
  • rand_angles()
  • compose_angles()
  • inverse_angles()
  • identity_quaternion()
  • rand_quaternion()
  • compose_quaternion()
  • inverse_quaternion()
  • rand_axis_angle()
  • compose_axis_angle()
  • matrix_x()
  • matrix_y()
  • matrix_z()
  • angles_to_matrix()
  • matrix_to_angles()
  • angles_to_quaternion()
  • matrix_to_quaternion()
  • axis_angle_to_quaternion()
  • quaternion_to_axis_angle()
  • matrix_to_axis_angle()
  • angles_to_axis_angle()
  • axis_angle_to_matrix()
  • quaternion_to_matrix()
  • quaternion_to_angles()
  • axis_angle_to_angles()
  • angles_to_xyz()
  • xyz_to_angles()
  • clebsch_gordan()
  • generators()
• Neural Network Functions
  • gate()
  • scalar_activation()
  • norm_activation()
• Functions for radial parameterization
  • soft_one_hot_linspace()
  • bessel()
  • poly_envelope()
  • soft_envelope()
• Haiku Modules
  • Linear
  • MultiLayerPerceptron
  • BatchNorm
  • Dropout
  • SymmetricTensorProduct
• Flax Modules
  • Linear
  • MultiLayerPerceptron
  • BatchNorm
• Signal on the Sphere
  • s2_irreps()
  • s2_dirac()
  • to_s2point()
  • to_s2grid()
  • from_s2grid()
  • SphericalSignal
• Extra Stuff
  • sus()
  • normalize_function()
  • scatter_sum()
  • scatter_mean()
  • scatter_max()
  • radius_graph()
• Utils
  • vmap()
  • equivariance_test()
  • assert_equivariant()
  • assert_output_dtype_matches_input_dtype()
• Legacy
  • FunctionalTensorProduct