wavelet scattering transform cosmology