Serial layers

Spherical harmonic transforms

RealSHT	Defines a module for computing the forward (real-valued) SHT.
InverseRealSHT	Defines a module for computing the inverse (real-valued) SHT.
RealVectorSHT	Defines a module for computing the forward (real) vector SHT.
InverseRealVectorSHT	Defines a module for computing the inverse (real-valued) vector SHT.

Convolutions

SpectralConvS2	Spectral convolution layer on \(S^2\) implemented via real SHT (Driscoll--Healy formulation, see https://api.semanticscholar.org/CorpusID:122817218).
DiscreteContinuousConvS2	Discrete-continuous (DISCO) convolution on the 2-sphere, as described in [1].
DiscreteContinuousConvTransposeS2	Discrete-continuous (DISCO) transpose convolution on the 2-sphere, as described in [1].

Filter basis

get_filter_basis
FilterBasis	Abstract base class for a filter basis
PiecewiseLinearFilterBasis	Tensor-product basis on a disk constructed from piecewise linear basis functions.
HarmonicFilterBasis	Hann-windowed Fourier basis on the disk: a Hann window in radius multiplied with a tensor-product sine/cosine basis in the Cartesian (x, y) coordinates on the disk.
ZernikeFilterBasis	Zernike polynomials which are defined on the disk.
FourierBesselFilterBasis	Fourier-Bessel (Disk Harmonic) filter basis on the unit disk.

Attention mechanism

AttentionS2	(Global) attention on the 2-sphere.
NeighborhoodAttentionS2	Neighborhood attention on the 2-sphere.

Resampling and quadrature

ResampleS2	Resampling module for signals on the 2-sphere \(S^2\).
QuadratureS2	Scalar quadrature on \(S^2\) for integrating spherical fields defined on a latitude/longitude grid.

Random fields

GaussianRandomFieldS2

Gaussian random field on the sphere via Karhunen--Loève expansion.