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

Add jax.Array to the index page

Browse Source
This commit is contained in:
yashkatariya 2022-12-01 15:38:38 -08:00
parent a6c7fa85bb
commit a2870a182e
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