just return the compiled func

Author: williambdean

pytensor/link/mlx/linker.py

@@ -35,6 +35,8 @@ def jit_compile(self, fn):
 
         inner_fn = mx.compile(fn)
 
+        return inner_fn
+
         def fn(*inputs, inner_fn=inner_fn):
             return inner_fn(*(mlx_typify(inp) for inp in inputs))