mirror of
https://github.com/ROCm/jax.git
synced 2025-04-18 12:56:07 +00:00
12 lines
188 B
ReStructuredText
12 lines
188 B
ReStructuredText
jax.distributed module
|
|
======================
|
|
|
|
.. currentmodule:: jax.distributed
|
|
|
|
.. automodule:: jax.distributed
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
initialize
|
|
shutdown |