Precompute Functions#

Spherical harmonic transforms.#

Function Name

Description

inverse()

Wrapper function around NumPy/JAX/Torch inverse methods

inverse_transform()

Inverse spherical harmonic transform (NumPy)

inverse_transform_jax()

Inverse spherical harmonic transform (JAX)

inverse_transform_torch()

Inverse spherical harmonic transform (Torch)

forward()

Wrapper function around NumPy/JAX/Torch forward methods

forward_transform()

Forward spherical harmonic transform (NumPy)

forward_transform_jax()

Forward spherical harmonic transform (JAX)

forward_transform_torch()

Forward spherical harmonic transform (Torch)

Wigner transforms.#

Function Name

Description

inverse()

Wrapper function around NumPy/JAX/Torch inverse methods

inverse_transform()

Inverse Wigner transform (NumPy)

inverse_transform_jax()

Inverse Wigner transform (JAX)

inverse_transform_torch()

Inverse Wigner transform (Torch)

forward()

Wrapper function around NumPy/JAX/Torch forward methods

forward_transform()

Forward Wigner transform (NumPy)

forward_transform_jax()

Forward Wigner transform (JAX)

forward_transform_torch()

Forward Wigner transform (Torch)

Fourier-Wigner transforms.#

Function Name

Description

inverse_transform()

Inverse Wigner transform with Fourier method (NumPy)

inverse_transform_jax()

Inverse Wigner transform with Fourier method (JAX)

forward_transform()

Forward Wigner transform with Fourier method (NumPy)

forward_transform_jax()

Forward Wigner transform with Fourier method (JAX)

Custom Operations#

Function Name

Description

wigner_subset_to_s2()

Transforms an arbitrary subset of Wigner coefficients onto a subset of spin signals on the sphere.

wigner_subset_to_s2_jax()

Transforms an arbitrary subset of Wigner coefficients onto a subset of spin signals on the sphere (JAX).

so3_to_wigner_subset()

Transforms a signal on the rotation group to an arbitrary subset of its Wigner coefficients.

so3_to_wigner_subset_jax()

Transforms a signal on the rotation group to an arbitrary subset of its Wigner coefficients (JAX).

s2_to_wigner_subset()

Transforms from a collection of arbitrary spin signals on the sphere to the corresponding collection of their harmonic coefficients.

s2_to_wigner_subset_jax()

Transforms from a collection of arbitrary spin signals on the sphere to the corresponding collection of their harmonic coefficients (JAX).

Constructing Kernels for precompute transforms.#

Function Name

Description

spin_spherical_kernel()

Builds a kernel including quadrature weights and Wigner-D coefficients for spherical harmonic transform.

wigner_kernel()

Builds a kernel including quadrature weights and Wigner-D coefficients for Wigner transform.

spin_spherical_kernel_jax()

Builds a kernel including quadrature weights and Wigner-D coefficients for spherical harmonic transform (JAX).

wigner_kernel_jax()

Builds a kernel including quadrature weights and Wigner-D coefficients for Wigner transform (JAX).

fourier_wigner_kernel()

Builds a kernel including quadrature weights and Fourier coefficienfs of Wigner d-functions

fourier_wigner_kernel_jax()

Builds a kernel including quadrature weights and Fourier coefficienfs of Wigner d-functions (JAX).

healpix_phase_shifts()

Builds a vector of corresponding phase shifts for each HEALPix latitudinal ring.