MLXBackend.set_device

MLXBackend.set_device(device_index)

Context manager that sets active compute device device for operations.

Parameters:

device_indexint

Index of the device to be set as active.