Load an MLX array from disk

Restores an array saved with mlx_save() and optionally places it on a specified device.

Usage

mlx_load(file, device = mlx_default_device())

Arguments

file

Path to a .npy file. The extension is appended automatically when missing.

device

Execution target: supply "gpu", "cpu", or an mlx_stream created via mlx_new_stream(). Default: mlx_default_device().

Value

An mlx array containing the file contents.

Details

Use an mlx_stream from mlx_new_stream() to load directly onto a specific stream; otherwise the array is placed on the current mlx_default_device().

See also

https://ml-explore.github.io/mlx/build/html/python/io.html#mlx.core.load