FINALLY BABY LETS PARTY! (IF YOU ARE READING THIS MAKE MORE PRs)

Author: cetagostini

pytensor/link/mlx/linker.py

@@ -35,8 +35,6 @@ 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))