mlx.core.stream

stream(s: Stream | ThreadLocalStream | Device | mlx.core.DeviceType) → mlx.core.StreamContext

Create a context manager to set the default device and stream.

Parameters:

s – The Stream or Device to set as the default.

Returns:

A context manager that sets the default device and stream.

Example: