Transforms

Spherical harmonic transforms.

Function Name	Description
inverse()	Wrapper function around NumPy/JAX inverse methods
inverse_numpy()	Inverse spherical harmonic transform (NumPy)
inverse_jax()	Inverse spherical harmonic transform (JAX)
forward()	Wrapper function around NumPy/JAX forward methods
forward_numpy()	Forward spherical harmonic transform (NumPy)
forward_jax()	Forward spherical harmonic transform (JAX)

Wigner transforms.

Function Name	Description
inverse()	Wrapper function around NumPy/JAX inverse methods
inverse_numpy()	Inverse Wigner transform (NumPy)
inverse_jax()	Inverse Wigner transform (JAX)
forward()	Wrapper function around NumPy/JAX forward methods
forward_numpy()	Forward Wigner transform (NumPy)
forward_jax()	Forward Wigner transform (JAX)

C/C++ backend gradient support

Function Name	Description
ssht_inverse()	Custom JAX frontend for inverse SSHT C spherical harmonic library.
ssht_forward()	Custom JAX frontend for forward SSHT C spherical harmonic library.
healpy_inverse()	Custom JAX frontend for inverse HEALPix C++ spherical harmonic library.
healpy_forward()	Custom JAX frontend for forwardHEALPix C++ spherical harmonic library.
inverse_jax_ssht()	Custom JAX frontend for hybrid inverse SSHT C Wigner transforms.
forward_jax_ssht()	Custom JAX frontend for hybrid forward SSHT C Wigner transforms.

On-the-fly Price-McEwen recursions.

Function Name	Description
inverse_latitudinal_step()	On-the-fly wigner-recursion for latitudinal portion of overall transform (NumPy)
inverse_latitudinal_step_jax()	On-the-fly wigner-recursion for latitudinal portion of overall transform (JAX)
forward_latitudinal_step()	On-the-fly wigner-recursion for latitudinal portion of overall transform (NumPy)
forward_latitudinal_step_jax()	On-the-fly wigner-recursion for latitudinal portion of overall transform (JAX)