update: matrix_power

Author: kozistr

pytorch_optimizer/__init__.py

@@ -22,4 +22,10 @@
 from pytorch_optimizer.sam import SAM
 from pytorch_optimizer.sgdp import SGDP
 from pytorch_optimizer.shampoo import Shampoo
-from pytorch_optimizer.utils import clip_grad_norm, get_optimizer_parameters, normalize_gradient, unit_norm
+from pytorch_optimizer.utils import (
+    clip_grad_norm,
+    get_optimizer_parameters,
+    matrix_power,
+    normalize_gradient,
+    unit_norm,
+)