Index _ | A | B | C | D | E | F | G | I | L | M | N | P | R | S | T | V | W _ __call__() (sun_diffusion.action.SUNPrincipalChiralAction method) (sun_diffusion.action.SUNToyAction method) (sun_diffusion.action.SUNToyPolynomialAction method) A adjoint() (in module sun_diffusion.linalg) B beta (sun_diffusion.action.SUNPrincipalChiralAction property) (sun_diffusion.action.SUNToyAction property) (sun_diffusion.action.SUNToyPolynomialAction property) C canonicalize_su2() (in module sun_diffusion.canon) canonicalize_su3() (in module sun_diffusion.canon) canonicalize_sun() (in module sun_diffusion.canon) casimir() (in module sun_diffusion.irreps) coeffs_to_group() (in module sun_diffusion.sun) compute_ess() (in module sun_diffusion.utils) compute_kl_div() (in module sun_diffusion.utils) D device_name() (in module sun_diffusion.devices) diffuse() (sun_diffusion.diffusion.VarianceExpandingDiffusion method) E embed_diag() (in module sun_diffusion.sun) embed_sun_algebra() (in module sun_diffusion.sun) extract_diag() (in module sun_diffusion.sun) extract_sun_algebra() (in module sun_diffusion.sun) F force_eigs() (sun_diffusion.action.SUNToyPolynomialAction method) G generate_partitions() (in module sun_diffusion.irreps) get_device() (in module sun_diffusion.devices) get_dtype() (in module sun_diffusion.devices) grab() (in module sun_diffusion.utils) grad_character() (in module sun_diffusion.irreps) group_to_coeffs() (in module sun_diffusion.sun) I inner_prod() (in module sun_diffusion.sun) L log_sun_hk() (in module sun_diffusion.heat) logsumexp() (in module sun_diffusion.utils) logsumexp_signed() (in module sun_diffusion.utils) M mat_angle() (in module sun_diffusion.sun) matrix_exp() (in module sun_diffusion.sun) matrix_log() (in module sun_diffusion.sun) module sun_diffusion.action sun_diffusion.canon sun_diffusion.devices sun_diffusion.diffusion sun_diffusion.gens sun_diffusion.heat sun_diffusion.irreps sun_diffusion.linalg sun_diffusion.sun sun_diffusion.utils N noise_coeff() (sun_diffusion.diffusion.PowerDiffusionSUN method) (sun_diffusion.diffusion.VarianceExpandingDiffusion method) (sun_diffusion.diffusion.VarianceExpandingDiffusionSUN method) P pauli() (in module sun_diffusion.gens) PowerDiffusionSUN (class in sun_diffusion.diffusion) proj_to_algebra() (in module sun_diffusion.sun) R random_sun_element() (in module sun_diffusion.sun) random_sun_lattice() (in module sun_diffusion.sun) random_un_haar_element() (in module sun_diffusion.sun) roll() (in module sun_diffusion.utils) S sample_sun_hk() (in module sun_diffusion.heat) set_device() (in module sun_diffusion.devices) set_dtype() (in module sun_diffusion.devices) sigma_func() (sun_diffusion.diffusion.PowerDiffusionSUN method) (sun_diffusion.diffusion.VarianceExpandingDiffusion method) (sun_diffusion.diffusion.VarianceExpandingDiffusionSUN method) summary() (in module sun_diffusion.devices) sun_diffusion.action module sun_diffusion.canon module sun_diffusion.devices module sun_diffusion.diffusion module sun_diffusion.gens module sun_diffusion.heat module sun_diffusion.irreps module sun_diffusion.linalg module sun_diffusion.sun module sun_diffusion.utils module sun_dual_hk() (in module sun_diffusion.heat) sun_hk() (in module sun_diffusion.heat) sun_score_dual_hk() (in module sun_diffusion.heat) sun_score_hk() (in module sun_diffusion.heat) sun_score_hk_autograd() (in module sun_diffusion.heat) sun_score_hk_autograd_v2() (in module sun_diffusion.heat) SUNPrincipalChiralAction (class in sun_diffusion.action) SUNToyAction (class in sun_diffusion.action) SUNToyPolynomialAction (class in sun_diffusion.action) T trace() (in module sun_diffusion.linalg) transpose() (in module sun_diffusion.linalg) V value_eigs() (sun_diffusion.action.SUNToyPolynomialAction method) VarianceExpandingDiffusion (class in sun_diffusion.diffusion) VarianceExpandingDiffusionSUN (class in sun_diffusion.diffusion) W weyl_character() (in module sun_diffusion.irreps) weyl_dimension() (in module sun_diffusion.irreps) wrap() (in module sun_diffusion.utils)