Back to Mlx

Devices and Streams

docs/src/python/devices_and_streams.rst

0.31.2352 B
Original Source

.. _devices_and_streams:

Devices and Streams

.. currentmodule:: mlx.core

.. autosummary:: :toctree: _autosummary

Device Stream default_device set_default_device default_stream new_stream new_thread_local_stream set_default_stream stream synchronize clear_streams device_count device_info