From a2870a182e1d935442362957087d92a34fa60ae5 Mon Sep 17 00:00:00 2001
From: yashkatariya <yashkatariya@google.com>
Date: Thu, 1 Dec 2022 15:38:38 -0800
Subject: [PATCH] Add jax.Array to the index page

---
 docs/index.rst | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/docs/index.rst b/docs/index.rst
index 37775784c..cbe6c2ff8 100644
--- a/docs/index.rst
+++ b/docs/index.rst
@@ -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