Index C | D | E | F | G | H | I | L | M | N | P | Q | R | S | T | U | W C compute_all_slices() (in module s2fft.recursions.price_mcewen) compute_all_slices_jax() (in module s2fft.recursions.price_mcewen) compute_eighth() (in module s2fft.recursions.trapani) compute_full() (in module s2fft.recursions.risbo) (in module s2fft.recursions.risbo_jax) (in module s2fft.recursions.trapani) (in module s2fft.recursions.turok) compute_full_jax() (in module s2fft.recursions.trapani) compute_full_loop() (in module s2fft.recursions.trapani) compute_full_vectorised() (in module s2fft.recursions.risbo) compute_full_vectorized() (in module s2fft.recursions.trapani) compute_quarter() (in module s2fft.recursions.trapani) (in module s2fft.recursions.turok) compute_quarter_jax() (in module s2fft.recursions.trapani) compute_quarter_slice() (in module s2fft.recursions.turok) compute_quarter_vectorized() (in module s2fft.recursions.trapani) compute_slice() (in module s2fft.recursions.turok) (in module s2fft.recursions.turok_jax) critical_log() (in module s2fft.logs) D debug_log() (in module s2fft.logs) downsample_by_two_mwss() (in module s2fft.utils.resampling) E elm2ind() (in module s2fft.sampling.s2_samples) elmn2ind() (in module s2fft.sampling.so3_samples) F f_shape() (in module s2fft.sampling.s2_samples) (in module s2fft.sampling.so3_samples) fill() (in module s2fft.recursions.turok) fill_eighth2quarter() (in module s2fft.recursions.trapani) fill_half2full() (in module s2fft.recursions.trapani) fill_half2full_jax() (in module s2fft.recursions.trapani) fill_half2full_vectorized() (in module s2fft.recursions.trapani) fill_quarter2half() (in module s2fft.recursions.trapani) fill_quarter2half_jax() (in module s2fft.recursions.trapani) fill_quarter2half_vectorized() (in module s2fft.recursions.trapani) flm_1d_to_2d() (in module s2fft.sampling.s2_samples) flm_2d_to_1d() (in module s2fft.sampling.s2_samples) flm_2d_to_hp() (in module s2fft.sampling.s2_samples) flm_hp_to_2d() (in module s2fft.sampling.s2_samples) flm_shape() (in module s2fft.sampling.s2_samples) flmn_1d_to_3d() (in module s2fft.sampling.so3_samples) flmn_3d_to_1d() (in module s2fft.sampling.so3_samples) flmn_shape() (in module s2fft.sampling.so3_samples) flmn_shape_1d() (in module s2fft.sampling.so3_samples) fnab_shape() (in module s2fft.sampling.so3_samples) forward() (in module s2fft.base_transforms.spherical) (in module s2fft.base_transforms.wigner) (in module s2fft.precompute_transforms.spherical) (in module s2fft.precompute_transforms.wigner) (in module s2fft.transforms.spherical) (in module s2fft.transforms.wigner) forward_jax() (in module s2fft.transforms.spherical) (in module s2fft.transforms.wigner) forward_jax_ssht() (in module s2fft.transforms.wigner) forward_latitudinal_step() (in module s2fft.transforms.otf_recursions) forward_latitudinal_step_jax() (in module s2fft.transforms.otf_recursions) forward_numpy() (in module s2fft.transforms.spherical) (in module s2fft.transforms.wigner) forward_transform() (in module s2fft.precompute_transforms.fourier_wigner) (in module s2fft.precompute_transforms.spherical) (in module s2fft.precompute_transforms.wigner) forward_transform_jax() (in module s2fft.precompute_transforms.fourier_wigner) (in module s2fft.precompute_transforms.spherical) (in module s2fft.precompute_transforms.wigner) forward_transform_torch() (in module s2fft.precompute_transforms.spherical) (in module s2fft.precompute_transforms.wigner) fourier_wigner_kernel() (in module s2fft.precompute_transforms.construct) fourier_wigner_kernel_jax() (in module s2fft.precompute_transforms.construct) ftm_shape() (in module s2fft.sampling.s2_samples) G generate_flm() (in module s2fft.utils.signal_generator) generate_flmn() (in module s2fft.utils.signal_generator) generate_precomputes() (in module s2fft.recursions.price_mcewen) generate_precomputes_jax() (in module s2fft.recursions.price_mcewen) generate_precomputes_wigner() (in module s2fft.recursions.price_mcewen) generate_precomputes_wigner_jax() (in module s2fft.recursions.price_mcewen) generate_rotate_dls() (in module s2fft.utils.rotation) H healpix_fft() (in module s2fft.utils.healpix_ffts) healpix_fft_cuda() (in module s2fft.utils.healpix_ffts) healpix_fft_jax() (in module s2fft.utils.healpix_ffts) healpix_fft_numpy() (in module s2fft.utils.healpix_ffts) healpix_fft_torch() (in module s2fft.utils.healpix_ffts) healpix_ifft() (in module s2fft.utils.healpix_ffts) healpix_ifft_cuda() (in module s2fft.utils.healpix_ffts) healpix_ifft_jax() (in module s2fft.utils.healpix_ffts) healpix_ifft_numpy() (in module s2fft.utils.healpix_ffts) healpix_ifft_torch() (in module s2fft.utils.healpix_ffts) healpix_phase_shifts() (in module s2fft.precompute_transforms.construct) hp_ang2pix() (in module s2fft.sampling.s2_samples) hp_getidx() (in module s2fft.sampling.s2_samples) I ind2elm() (in module s2fft.sampling.s2_samples) info_log() (in module s2fft.logs) init() (in module s2fft.recursions.trapani) init_jax() (in module s2fft.recursions.trapani) init_nonjax() (in module s2fft.recursions.trapani) inverse() (in module s2fft.base_transforms.spherical) (in module s2fft.base_transforms.wigner) (in module s2fft.precompute_transforms.spherical) (in module s2fft.precompute_transforms.wigner) (in module s2fft.transforms.spherical) (in module s2fft.transforms.wigner) inverse_jax() (in module s2fft.transforms.spherical) (in module s2fft.transforms.wigner) inverse_jax_ssht() (in module s2fft.transforms.wigner) inverse_latitudinal_step() (in module s2fft.transforms.otf_recursions) inverse_latitudinal_step_jax() (in module s2fft.transforms.otf_recursions) inverse_numpy() (in module s2fft.transforms.spherical) (in module s2fft.transforms.wigner) inverse_transform() (in module s2fft.precompute_transforms.fourier_wigner) (in module s2fft.precompute_transforms.spherical) (in module s2fft.precompute_transforms.wigner) inverse_transform_jax() (in module s2fft.precompute_transforms.fourier_wigner) (in module s2fft.precompute_transforms.spherical) (in module s2fft.precompute_transforms.wigner) inverse_transform_torch() (in module s2fft.precompute_transforms.spherical) (in module s2fft.precompute_transforms.wigner) L lm2lm_hp() (in module s2fft.sampling.s2_samples) M module s2fft.base_transforms.spherical s2fft.base_transforms.wigner s2fft.logs s2fft.precompute_transforms.construct s2fft.precompute_transforms.custom_ops s2fft.precompute_transforms.fourier_wigner s2fft.precompute_transforms.spherical s2fft.precompute_transforms.wigner s2fft.recursions.price_mcewen s2fft.recursions.risbo s2fft.recursions.risbo_jax s2fft.recursions.trapani s2fft.recursions.turok s2fft.recursions.turok_jax s2fft.sampling.s2_samples s2fft.sampling.so3_samples s2fft.transforms.c_backend_spherical s2fft.transforms.otf_recursions s2fft.transforms.spherical s2fft.transforms.wigner s2fft.utils s2fft.utils.healpix_ffts s2fft.utils.quadrature s2fft.utils.quadrature_jax s2fft.utils.quadrature_torch s2fft.utils.resampling s2fft.utils.resampling_jax s2fft.utils.resampling_torch s2fft.utils.rotation s2fft.utils.signal_generator mw_to_mwss() (in module s2fft.utils.resampling) (in module s2fft.utils.resampling_jax) (in module s2fft.utils.resampling_torch) mw_to_mwss_phi() (in module s2fft.utils.resampling) (in module s2fft.utils.resampling_jax) (in module s2fft.utils.resampling_torch) mw_to_mwss_theta() (in module s2fft.utils.resampling) (in module s2fft.utils.resampling_jax) (in module s2fft.utils.resampling_torch) mw_weights() (in module s2fft.utils.quadrature) (in module s2fft.utils.quadrature_jax) (in module s2fft.utils.quadrature_torch) N ncoeff() (in module s2fft.sampling.s2_samples) nphi_equiang() (in module s2fft.sampling.s2_samples) nphi_equitorial_band() (in module s2fft.sampling.s2_samples) nphi_ring() (in module s2fft.sampling.s2_samples) ntheta() (in module s2fft.sampling.s2_samples) ntheta_extension() (in module s2fft.sampling.s2_samples) P p2phi_equiang() (in module s2fft.sampling.s2_samples) p2phi_ring() (in module s2fft.sampling.s2_samples) p2phi_rings() (in module s2fft.utils.healpix_ffts) p2phi_rings_jax() (in module s2fft.utils.healpix_ffts) periodic_extension() (in module s2fft.utils.resampling) (in module s2fft.utils.resampling_jax) (in module s2fft.utils.resampling_torch) periodic_extension_spatial_mwss() (in module s2fft.utils.resampling) (in module s2fft.utils.resampling_jax) (in module s2fft.utils.resampling_torch) phis_equiang() (in module s2fft.sampling.s2_samples) phis_ring() (in module s2fft.sampling.s2_samples) Q quad_weight_dh_theta_only() (in module s2fft.utils.quadrature) (in module s2fft.utils.quadrature_jax) (in module s2fft.utils.quadrature_torch) quad_weights() (in module s2fft.utils.quadrature) (in module s2fft.utils.quadrature_jax) (in module s2fft.utils.quadrature_torch) quad_weights_dh() (in module s2fft.utils.quadrature) (in module s2fft.utils.quadrature_jax) (in module s2fft.utils.quadrature_torch) quad_weights_gl() (in module s2fft.utils.quadrature) (in module s2fft.utils.quadrature_jax) (in module s2fft.utils.quadrature_torch) quad_weights_hp() (in module s2fft.utils.quadrature) (in module s2fft.utils.quadrature_jax) (in module s2fft.utils.quadrature_torch) quad_weights_mw() (in module s2fft.utils.quadrature) (in module s2fft.utils.quadrature_jax) (in module s2fft.utils.quadrature_torch) quad_weights_mw_theta_only() (in module s2fft.utils.quadrature) (in module s2fft.utils.quadrature_jax) (in module s2fft.utils.quadrature_torch) quad_weights_mwss() (in module s2fft.utils.quadrature) (in module s2fft.utils.quadrature_jax) (in module s2fft.utils.quadrature_torch) quad_weights_mwss_theta_only() (in module s2fft.utils.quadrature) (in module s2fft.utils.quadrature_jax) (in module s2fft.utils.quadrature_torch) quad_weights_transform() (in module s2fft.utils.quadrature) (in module s2fft.utils.quadrature_jax) (in module s2fft.utils.quadrature_torch) R reindex() (in module s2fft.recursions.turok_jax) ring_phase_shift_hp() (in module s2fft.sampling.s2_samples) ring_phase_shifts_hp() (in module s2fft.utils.healpix_ffts) ring_phase_shifts_hp_jax() (in module s2fft.utils.healpix_ffts) rotate_flms() (in module s2fft.utils.rotation) S s2_to_wigner_subset() (in module s2fft.precompute_transforms.custom_ops) s2_to_wigner_subset_jax() (in module s2fft.precompute_transforms.custom_ops) s2fft.base_transforms.spherical module s2fft.base_transforms.wigner module s2fft.logs module s2fft.precompute_transforms.construct module s2fft.precompute_transforms.custom_ops module s2fft.precompute_transforms.fourier_wigner module s2fft.precompute_transforms.spherical module s2fft.precompute_transforms.wigner module s2fft.recursions.price_mcewen module s2fft.recursions.risbo module s2fft.recursions.risbo_jax module s2fft.recursions.trapani module s2fft.recursions.turok module s2fft.recursions.turok_jax module s2fft.sampling.s2_samples module s2fft.sampling.so3_samples module s2fft.transforms.c_backend_spherical module s2fft.transforms.otf_recursions module s2fft.transforms.spherical module s2fft.transforms.wigner module s2fft.utils module s2fft.utils.healpix_ffts module s2fft.utils.quadrature module s2fft.utils.quadrature_jax module s2fft.utils.quadrature_torch module s2fft.utils.resampling module s2fft.utils.resampling_jax module s2fft.utils.resampling_torch module s2fft.utils.rotation module s2fft.utils.signal_generator module setup_logging() (in module s2fft.logs) so3_to_wigner_subset() (in module s2fft.precompute_transforms.custom_ops) so3_to_wigner_subset_jax() (in module s2fft.precompute_transforms.custom_ops) spectral_folding() (in module s2fft.utils.healpix_ffts) spectral_folding_jax() (in module s2fft.utils.healpix_ffts) spectral_folding_torch() (in module s2fft.utils.healpix_ffts) spectral_periodic_extension() (in module s2fft.utils.healpix_ffts) spectral_periodic_extension_jax() (in module s2fft.utils.healpix_ffts) spectral_periodic_extension_torch() (in module s2fft.utils.healpix_ffts) spin_spherical_kernel() (in module s2fft.precompute_transforms.construct) spin_spherical_kernel_jax() (in module s2fft.precompute_transforms.construct) T t2theta() (in module s2fft.sampling.s2_samples) thetas() (in module s2fft.sampling.s2_samples) U unextend() (in module s2fft.utils.resampling) (in module s2fft.utils.resampling_jax) (in module s2fft.utils.resampling_torch) upsample_by_two_mwss() (in module s2fft.utils.resampling) (in module s2fft.utils.resampling_jax) (in module s2fft.utils.resampling_torch) upsample_by_two_mwss_ext() (in module s2fft.utils.resampling) (in module s2fft.utils.resampling_jax) (in module s2fft.utils.resampling_torch) W warning_log() (in module s2fft.logs) wigner_kernel() (in module s2fft.precompute_transforms.construct) wigner_kernel_jax() (in module s2fft.precompute_transforms.construct) wigner_subset_to_s2() (in module s2fft.precompute_transforms.custom_ops) wigner_subset_to_s2_jax() (in module s2fft.precompute_transforms.custom_ops)
