add a liberal for NS coefficient

skyw · skyw · commit fc589ee2bffd · 2025-12-22T12:53:11.000-08:00
Signed-off-by: Hao Wu &lt;skyw@nvidia.com&gt;
diff --git a/emerging_optimizers/orthogonalized_optimizers/muon_utils.py b/emerging_optimizers/orthogonalized_optimizers/muon_utils.py
@@ -20,7 +20,7 @@
 from emerging_optimizers import triton_kernels
 
 
-__all__ = ["newton_schulz", "newton_schulz_tp"]
+__all__ = ["newton_schulz", "newton_schulz_tp", "NSCoeffT"]
 
 NSCoeffT = Literal["simple", "quintic", "polar_express", "aol", "custom"]
 
