mlx.core.ThreadLocalStream

mlx.core.ThreadLocalStream#

class ThreadLocalStream#

A stream that will be unique per thread and can be used to run operations on a given device.

__init__(*args, **kwargs)#

Methods

__init__(*args, **kwargs)

Attributes

device

(self) -> mlx.core.Device