mlx.core.Device

class Device(*args, **kwargs)

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