Merge branch 'master' into fix/bf16-zero3-quantized-weights

Author: juyterman1000

deepspeed/runtime/zero/utils.py

@@ -47,7 +47,7 @@ class ZeRORuntimeException(Exception):
 
 # Add MuonWithAuxAdam to supported list if muon is installed
 try:
-    from deepspeed.runtime.muon_optimizer import MuonWithAuxAdam
+    from deepspeed.runtime.zero.muon.muon_optimizer import MuonWithAuxAdam
     ZERO_SUPPORTED_OPTIMIZERS.append(MuonWithAuxAdam)
 except ImportError:
     pass