Skip to content

Commit e58af3e

Browse files
committed
Create MLX branch
1 parent e062d3b commit e58af3e

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

keras/backend/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,10 @@
4040
elif backend() == "numpy":
4141
from keras.backend.numpy import * # noqa: F403
4242

43+
distribution_lib = None
44+
elif backend() == "mlx":
45+
from keras.backend.mlx import * # noqa: F403
46+
4347
distribution_lib = None
4448
else:
4549
raise ValueError(f"Unable to import backend : {backend()}")

0 commit comments

Comments
 (0)