mirror of
https://github.com/ROCm/jax.git
synced 2025-04-14 10:56:06 +00:00
20 lines
453 B
ReStructuredText
20 lines
453 B
ReStructuredText
``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
|
|
|