jax.distributed.rst

jax.distributed module

.. currentmodule:: jax.distributed

.. automodule:: jax.distributed

.. autosummary::
    :toctree: _autosummary

    initialize
    shutdown