mlx.core.Device

class Device

A device to run operations on.

__init__(self, type: mlx.core.DeviceType, index: int = 0) → None

Methods

__init__(self, type[, index])

Attributes

type	(self) -> mlx.core.DeviceType