feat(pt): add AdaMuon optimizer

(cherry picked from commit 81d66ff)

Author: OutisLi

deepmd/pt/optimizer/__init__.py

@@ -1,9 +1,12 @@
 # SPDX-License-Identifier: LGPL-3.0-or-later
+from .adamuon import (
+    AdaMuonOptimizer,
+)
 from .KFWrapper import (
     KFOptimizerWrapper,
 )
 from .LKF import (
     LKFOptimizer,
 )
 
-__all__ = ["KFOptimizerWrapper", "LKFOptimizer"]
+__all__ = ["AdaMuonOptimizer", "KFOptimizerWrapper", "LKFOptimizer"]