feat(linalg): support eigh on mlx (#20843)

lpizzinidev · web-flow · commit 6a957bcf4a7e · 2025-02-02T10:19:23.000-08:00
diff --git a/keras/src/backend/mlx/linalg.py b/keras/src/backend/mlx/linalg.py
@@ -18,7 +18,8 @@ def eig(a):
 
 
 def eigh(a):
-    raise NotImplementedError("eigh not yet implemented in mlx.")
+    with mx.stream(mx.cpu):
+        return mx.linalg.eigh(a)
 
 
 def lu_factor(a):
