https://github.com/google/jax
Raw File
Tip revision: 4cb5eeee537d044911228870bf7b1d38ea478e2c authored by jax authors on 03 October 2023, 20:33:22 UTC
Merge pull request #17905 from skye:version
Tip revision: 4cb5eee
jax.distributed.rst
``jax.distributed`` module
==========================

.. currentmodule:: jax.distributed

.. automodule:: jax.distributed

.. autosummary::
    :toctree: _autosummary

    initialize
    shutdown
back to top