mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 04:16:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/docs/jax.experimental.multihost_utils.rst

20 lines
453 B
ReStructuredText
Raw Normal View History

Document jax.experimental.multihost_utils
2023-05-04 13:50:29 -07:00
``jax.experimental.multihost_utils`` module
Fix a sphinx error.
2023-05-05 09:03:39 -07:00
===========================================
Document jax.experimental.multihost_utils
2023-05-04 13:50:29 -07:00
.. 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
Reference in New Issue Copy Permalink