mlx.core.set_default_device

set_default_device(device: Device | DeviceType) → None

Set the default device.