Precompute Functions#
Function Name |
Description |
---|---|
Wrapper function around NumPy/JAX/Torch inverse methods |
|
Inverse spherical harmonic transform (NumPy) |
|
Inverse spherical harmonic transform (JAX) |
|
Inverse spherical harmonic transform (Torch) |
|
Wrapper function around NumPy/JAX/Torch forward methods |
|
Forward spherical harmonic transform (NumPy) |
|
Forward spherical harmonic transform (JAX) |
|
Forward spherical harmonic transform (Torch) |
Function Name |
Description |
---|---|
Wrapper function around NumPy/JAX/Torch inverse methods |
|
Inverse Wigner transform (NumPy) |
|
Inverse Wigner transform (JAX) |
|
Inverse Wigner transform (Torch) |
|
Wrapper function around NumPy/JAX/Torch forward methods |
|
Forward Wigner transform (NumPy) |
|
Forward Wigner transform (JAX) |
|
Forward Wigner transform (Torch) |
Function Name |
Description |
---|---|
Inverse Wigner transform with Fourier method (NumPy) |
|
Inverse Wigner transform with Fourier method (JAX) |
|
Forward Wigner transform with Fourier method (NumPy) |
|
Forward Wigner transform with Fourier method (JAX) |
Function Name |
Description |
---|---|
Transforms an arbitrary subset of Wigner coefficients onto a subset of spin signals on the sphere. |
|
Transforms an arbitrary subset of Wigner coefficients onto a subset of spin signals on the sphere (JAX). |
|
Transforms a signal on the rotation group to an arbitrary subset of its Wigner coefficients. |
|
Transforms a signal on the rotation group to an arbitrary subset of its Wigner coefficients (JAX). |
|
Transforms from a collection of arbitrary spin signals on the sphere to the corresponding collection of their harmonic coefficients. |
|
Transforms from a collection of arbitrary spin signals on the sphere to the corresponding collection of their harmonic coefficients (JAX). |
Function Name |
Description |
---|---|
Builds a kernel including quadrature weights and Wigner-D coefficients for spherical harmonic transform. |
|
Builds a kernel including quadrature weights and Wigner-D coefficients for Wigner transform. |
|
Builds a kernel including quadrature weights and Wigner-D coefficients for spherical harmonic transform (JAX). |
|
Builds a kernel including quadrature weights and Wigner-D coefficients for Wigner transform (JAX). |
|
Builds a kernel including quadrature weights and Fourier coefficienfs of Wigner d-functions |
|
Builds a kernel including quadrature weights and Fourier coefficienfs of Wigner d-functions (JAX). |
|
Builds a vector of corresponding phase shifts for each HEALPix latitudinal ring. |