Index A | C | E | F | G | I | M | N | O | P | S | U A adaptive_early_exit_criteria() (in module emerging_optimizers.utils.eig) C calculate_adam_update() (in module emerging_optimizers.scalar_optimizers) calculate_ademamix_update() (in module emerging_optimizers.scalar_optimizers) calculate_laprop_update() (in module emerging_optimizers.scalar_optimizers) calculate_lion_update() (in module emerging_optimizers.scalar_optimizers) calculate_signum_update() (in module emerging_optimizers.scalar_optimizers) calculate_sim_ademamix_update() (in module emerging_optimizers.scalar_optimizers) E eig_orthogonal_iteration() (in module emerging_optimizers.utils.eig) eigh_with_fallback() (in module emerging_optimizers.utils.eig) emerging_optimizers.orthogonalized_optimizers module emerging_optimizers.orthogonalized_optimizers.muon_utils module emerging_optimizers.scalar_optimizers module emerging_optimizers.soap.soap module emerging_optimizers.utils module emerging_optimizers.utils.eig module F fp32_matmul_precision() (in module emerging_optimizers.utils) G get_pg_rank() (in module emerging_optimizers.utils) get_pg_size() (in module emerging_optimizers.utils) I init_kronecker_factors() (in module emerging_optimizers.soap.soap) M module emerging_optimizers.orthogonalized_optimizers emerging_optimizers.orthogonalized_optimizers.muon_utils emerging_optimizers.scalar_optimizers emerging_optimizers.soap.soap emerging_optimizers.utils emerging_optimizers.utils.eig Muon (class in emerging_optimizers.orthogonalized_optimizers) N newton_schulz() (in module emerging_optimizers.orthogonalized_optimizers.muon_utils) newton_schulz_step() (in module emerging_optimizers.orthogonalized_optimizers.muon_utils) newton_schulz_tp() (in module emerging_optimizers.orthogonalized_optimizers.muon_utils) O orthogonalize() (emerging_optimizers.orthogonalized_optimizers.OrthogonalizedOptimizer method) OrthogonalizedOptimizer (class in emerging_optimizers.orthogonalized_optimizers) P precondition() (in module emerging_optimizers.soap.soap) S SOAP (class in emerging_optimizers.soap.soap) step() (emerging_optimizers.orthogonalized_optimizers.OrthogonalizedOptimizer method) (emerging_optimizers.soap.soap.SOAP method) U update_eigenbasis_and_momentum() (in module emerging_optimizers.soap.soap) update_kronecker_factors() (in module emerging_optimizers.soap.soap)
