https://github.com/google/jax
Raw File
Tip revision: e034432872545c68c91f598e84c20ffd4c7c6cea authored by jax authors on 26 September 2022, 17:04:14 UTC
Merge pull request #12513 from inoryy:patch-4
Tip revision: e034432
jax.distributed.rst
jax.distributed module
======================

.. currentmodule:: jax.distributed

.. automodule:: jax.distributed

.. autosummary::
    :toctree: _autosummary

    initialize
    shutdown
back to top