Core Functions#

Generative models#

Function Name

Description

build_encoder()

Builds a scattering covariance encoding function (latent encoder).

build_generator()

Builds a scattering covariance generator function. (generative model).

Scattering transforms#

Function Name

Description

scatter()

Compute directional scattering covariances on the sphere (Mousset et al 2024).

scatter_c()

Compute directional scattering covariances on the sphere using a custom C backend (Mousset et al 2024).

Compression functions#

Function Name

Description

C01_C11_to_isotropic()

Convert scattering covariances to their isotropic counterpart.

Optimisation functions#

Function Name

Description

fit_optax()

Minimises the declared loss function starting at params using optax (adam).

l2_covariance_loss()

L2 loss wrapper for the scattering covariance.

l2_loss()

L2 loss for a single scattering covariance.

get_P00prime()

Computes P00prime which is the averaged power within each wavelet scale.