Revision 7b182ec8791c97ba61bbe186e956e7bdd340a0f0 authored by Brian Patton on 11 March 2024, 16:05:25 UTC, committed by jax authors on 11 March 2024, 16:56:55 UTC
PiperOrigin-RevId: 614684199
1 parent 71ec6e3
jax.experimental.multihost_utils.rst
``jax.experimental.multihost_utils`` module
===========================================
.. automodule:: jax.experimental.multihost_utils
.. currentmodule:: jax.experimental.multihost_utils
Multihost Utils API Reference
-----------------------------
.. autosummary::
:toctree: _autosummary
broadcast_one_to_all
sync_global_devices
process_allgather
assert_equal
host_local_array_to_global_array
global_array_to_host_local_array
![swh spinner](/static/img/swh-spinner.gif)
Computing file changes ...