mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #13484 from google:index

Browse Source 
PiperOrigin-RevId: 492323132
This commit is contained in:
jax authors 2022-12-01 16:02:50 -08:00
commit ed9519dadf
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
docs/index.rst
View File

@ -5,6 +5,11 @@ JAX is Autograd_ and XLA_, brought together for high-performance numerical compu
It provides composable transformations of Python+NumPy programs: differentiate, vectorize,
parallelize, Just-In-Time compile to GPU/TPU, and more.
.. note::
JAX 0.4.0 introduces new parallelism APIs, including breaking changes to :func:`jax.experimental.pjit` and a new unified ``jax.Array`` type.
Please see `Parallelism with JAX <https://jax.readthedocs.io/en/latest/notebooks/Parallelism_with_JAX.html>`_ tutorial and the :ref:`jax-array-migration`
guide for more information.
.. toctree::
:maxdepth: 1
:caption: Getting Started