mlx.core.export_function

export_function(file: str, fun: Callable, *args, shapeless: bool = False, **kwargs) → None

Export a function to a file.

Example input arrays must be provided to export a function. The example inputs can be variable *args and **kwargs or a tuple of arrays and/or dictionary of string keys with array values.

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) – File path to export the function to.

• fun (Callable) – A function which takes as input zero or more array and returns one or more array.

• *args (array) – Example array inputs to the function.

• shapeless (bool, optional) – Whether or not the function allows inputs with variable shapes. Default: False.

• **kwargs (array) – Additional example keyword array inputs to the function.

Example

def fun(x, y):
    return x + y

x = mx.array(1)
y = mx.array([1, 2, 3])
mx.export_function("fun.mlxfn", fun, x, y=y)