rocm_jax/docs/jax.distributed.rst

``jax.distributed`` module
==========================

.. currentmodule:: jax.distributed

.. automodule:: jax.distributed

.. autosummary::
    :toctree: _autosummary

    initialize
    shutdown
DOC: make API doc titles more uniform 2023-01-18 10:59:42 -08:00			``jax.distributed`` module
			`==========================`
Add jax.distributed and jax.dlpack to the docs. Reorder the doc modules into something closer to alphabetical order. Add missing functions from jax.scipy.linalg and jax.scipy.signal to the docs. 2022-02-17 16:05:33 -05:00
			`.. currentmodule:: jax.distributed`

			`.. automodule:: jax.distributed`

			`.. autosummary::`
			`:toctree: _autosummary`

Document multiprocess GPU support. Fixes #2731 2022-08-08 17:47:34 -04:00			`initialize`
			`shutdown`