1
0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00

Add cross-linking for the migration guide and the parallelism with JAX

tutorial

Co-authored-by: Skye Wanderman-Milne <skyewm@google.com>
This commit is contained in:
yashkatariya 2022-12-01 14:27:45 -08:00
parent 60a45a400b
commit 70d50814b1
3 changed files with 7 additions and 0 deletions

@ -1,3 +1,4 @@
(jax-array-migration)=
# jax.Array migration
**yashkatariya@**
@ -17,6 +18,8 @@ the unified jax.Array
After the migration is complete `jax.Array` will be the only type of array in
JAX.
This doc explains how to migrate existing codebases to `jax.Array`. For more information on using `jax.Array` and JAX parallelism APIs, see the [Parallelism with JAX](https://jax.readthedocs.io/en/latest/notebooks/Parallelism_with_JAX.html) tutorial.
### How to enable jax.Array?

@ -17,6 +17,8 @@
"source": [
"**This tutorial discusses parallelism via `jax.Array`, the unified array object model available in JAX v0.4.0 and newer.**\n",
"\n",
"See {ref}`jax-array-migration` guide for migrating existing pre-v0.4.0 codebases to `jax.Array`.\n",
"\n",
"**The features required by `jax.Array` are not supported by the Colab TPU runtime at this time.**"
]
},

@ -19,6 +19,8 @@ kernelspec:
**This tutorial discusses parallelism via `jax.Array`, the unified array object model available in JAX v0.4.0 and newer.**
See {ref}`jax-array-migration` guide for migrating existing pre-v0.4.0 codebases to `jax.Array`.
**The features required by `jax.Array` are not supported by the Colab TPU runtime at this time.**
```{code-cell}