mlx.core.import_function

import_function(file: str) → Callable

Import a function from a file.

The imported function can be called either with *args and **kwargs or with a tuple of arrays and/or dictionary of string keys with array values. Imported functions always return a tuple of arrays.

Warning

This is part of an experimental API which is likely to change in future versions of MLX. Functions exported with older versions of MLX may not be compatible with future versions.

Parameters:

file (str) – The file path to import the function from.

Returns:

The imported function.

Return type:

Callable

Example

>>> fn = mx.import_function("function.mlxfn")
>>> out = fn(a, b, x=x, y=y)[0]
>>>
>>> out = fn((a, b), {"x": x, "y": y}[0]