s2fft is currently in an open beta, please provide feedback on GitHub

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)

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.

healpix_phase_shifts()

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