Skip to content

Commit 4c90dfb

Browse files
authored
feat(math): support erfinv on mlx (#19628)
1 parent 722883a commit 4c90dfb

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

keras/src/backend/mlx/math.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -102,6 +102,11 @@ def erf(x):
102102
return mx.erf(x)
103103

104104

105+
def erfinv(x):
106+
x = convert_to_tensor(x)
107+
return mx.erfinv(x)
108+
109+
105110
def solve(a, b):
106111
raise NotImplementedError(
107112
"Linear system solving not yet implemented in mlx"

keras/src/backend/mlx/trainer.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@
33

44
from keras.src import backend
55
from keras.src import callbacks as callbacks_module
6-
from keras.src import ops
76
from keras.src import optimizers as optimizers_module
87
from keras.src import tree
98
from keras.src.backend.common import standardize_dtype

0 commit comments

Comments
 (0)