https://github.com/google/jax
Raw File
Tip revision: 500da57e91eb4b9094eb2b5cfa0826691862417f authored by jax authors on 07 May 2024, 14:07:04 UTC
Merge pull request #21077 from merrymercy:patch-1
Tip revision: 500da57
jax.distributed.rst
``jax.distributed`` module
==========================

.. currentmodule:: jax.distributed

.. automodule:: jax.distributed

.. autosummary::
    :toctree: _autosummary

    initialize
    shutdown
back to top