simple mlx fft implementation (#20777)

acsweet · web-flow · commit 603affa5e783 · 2025-01-17T13:59:21.000-08:00
diff --git a/keras/src/backend/mlx/math.py b/keras/src/backend/mlx/math.py
@@ -55,8 +55,8 @@ def extract_sequences(x, sequence_length, sequence_stride):
 
 
 def fft(x):
-    # TODO: https://ml-explore.github.io/mlx/build/html/python/fft.html#fft
-    raise NotImplementedError("fft not yet implemented in mlx")
+    x = convert_to_tensor(x)
+    return mx.fft(x)
 
 
 def fft2(x):
