mlx.core.Device

class mlx.core.Device

A device to run operations on.

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

Methods

__init__(self, type[, index])

Attributes

type