mlx/docs/build/html/_sources/python/distributed.rst

23 lines
426 B
ReStructuredText
Raw Normal View History

2024-06-07 11:28:06 +08:00
.. _distributed:
.. currentmodule:: mlx.core.distributed
Distributed Communication
==========================
MLX provides a distributed communication package using MPI. The MPI library is
loaded at runtime; if MPI is available then distributed communication is also
made available.
.. autosummary::
:toctree: _autosummary
Group
is_available
init
all_sum
all_gather
2024-09-18 03:06:14 +08:00
send
recv
recv_like