mlx.core.device_info

Contents

mlx.core.device_info#

device_info(d: Device = Device(cpu, 0)) dict[str, str | int]#

Get information about a device.

Returns a dictionary with device properties. Available keys depend on the backend and device type. Common keys include device_name, architecture, and total_memory (or memory_size).

Parameters:

d (Device) – The device to query (defaults to the default device).

Returns:

Device information.

Return type:

dict