Add NAMO and NAMO-D optimizer configs (#2966)

Author: suraj-ranganath

lib/levanter/src/levanter/optim/__init__.py

@@ -25,6 +25,8 @@
     "MuonHConfig",
     "ScaleByMuonState",
     "GrugMuonConfig",
+    "NamoConfig",
+    "NamoDConfig",
     # rmsprop
     "RMSPropMomentumConfig",
     "ScaleByRMSPropMomState",
@@ -50,6 +52,7 @@
 from .muon import MuonConfig, ScaleByMuonState
 from .grugmuon import GrugMuonConfig
 from .muonh import MuonHConfig
+from .namo import NamoConfig, NamoDConfig
 from .rmsprop import RMSPropMomentumConfig, ScaleByRMSPropMomState
 from .scion import ScaleByScionState, ScionConfig
 from .soap import SoapConfig