e3nn_mlx.o3.spherical_harmonics

e3nn_mlx.o3.spherical_harmonics(degrees, vectors, *, normalize=True, normalization='component', use_custom_kernel=True)[source]

Evaluate real spherical harmonics in the canonical e3nn basis.

A raw MLX input returns a raw MLX array. An IrrepsArray input returns an IrrepsArray and uses its vector parity.

Parameters:

  • degrees (int | Sequence[int] | Irreps | str)

  • normalize (bool)

  • normalization (str)

  • use_custom_kernel (bool)