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

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)