https://github.com/google/jax
Raw File
Tip revision: b868cf7c07e25fc214c0f2f311335bc61d90c265 authored by jax authors on 12 December 2022, 18:03:54 UTC
Merge pull request #13616 from jakevdp:fix-sparse-error
Tip revision: b868cf7
jax.distributed.rst
jax.distributed module
======================

.. currentmodule:: jax.distributed

.. automodule:: jax.distributed

.. autosummary::
    :toctree: _autosummary

    initialize
    shutdown
back to top