Skip to content

Commit 2bc4baf

Browse files
authored
implement remaining functions in mlx.numpy.py (#20816)
* mlx.numpy.py implementation * formatting
1 parent bc2c6a3 commit 2bc4baf

File tree

3 files changed

+336
-125
lines changed

3 files changed

+336
-125
lines changed

keras/src/backend/mlx/linalg.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,10 @@
77

88

99
def det(a):
10-
raise NotImplementedError("det not yet implemented in mlx.")
10+
# TODO: Swap to mlx.linalg.det when supported
11+
a = jnp.array(a)
12+
output = jnp.linalg.det(a)
13+
return mx.array(output)
1114

1215

1316
def eig(a):

0 commit comments

Comments
 (0)