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

Wavelet Transforms#

Numpy transforms#

Function Name

Description

synthesis_looped()

Loopy implementation of mapping from wavelet to pixel space.

synthesis()

Vectorised implementation of mapping from wavelet to pixel space.

analysis_looped()

Loopy implementation of mapping from pixel to wavelet space.

analysis()

Vectorised implementation of mapping from pixel to wavelet space.

JAX transforms#

Function Name

Description

synthesis()

JAX implementation of mapping from wavelet to pixel space (Recursive).

analysis()

JAX implementation of mapping from pixel to wavelet space (Recursive).

flm_to_analysis()

JAX implementation of mapping from harmonic to wavelet coefficients only (Recursive).

synthesis()

JAX implementation of mapping from wavelet to pixel space (fully precompute).

analysis()

JAX implementation of mapping from pixel to wavelet space (fully precompute).

flm_to_analysis()

JAX implementation of mapping from harmonic to wavelet coefficients only (fully precompute).

PyTorch transforms#

Function Name

Description

synthesis()

PyTorch implementation of mapping from wavelet to pixel space (fully precompute).

analysis()

PyTorch implementation of mapping from pixel to wavelet space (fully precompute).

flm_to_analysis()

PyTorch implementation of mapping from harmonic to wavelet coefficients only (fully precompute).

Matrices precomputations#

Function Name

Description

generate_wigner_precomputes()

JAX/PyTorch function to generate precompute arrays for underlying Wigner transforms.

generate_full_precomputes()

JAX/PyTorch function to generate precompute arrays for fully precompute transforms.