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

Merge pull request #26082 from Saransh-cpp:index-update-syntax-link

Browse Source 
PiperOrigin-RevId: 719356167
This commit is contained in:
jax authors 2025-01-24 10:37:07 -08:00
commit 726abc9c31
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/notebooks/thinking_in_jax.ipynb
View File

@ -248,7 +248,7 @@
"id": "yRYF0YgO3F4H"
},
"source": [
"For updating individual elements, JAX provides an [indexed update syntax](https://jax.readthedocs.io/en/latest/jax.ops.html#indexed-update-operators) that returns an updated copy:"
"For updating individual elements, JAX provides an [indexed update syntax](https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.ndarray.at.html#jax-numpy-ndarray-at) that returns an updated copy:"
]
},
{

2
docs/notebooks/thinking_in_jax.md
View File

@ -117,7 +117,7 @@ x[0] = 10
+++ {"id": "yRYF0YgO3F4H"}
For updating individual elements, JAX provides an [indexed update syntax](https://jax.readthedocs.io/en/latest/jax.ops.html#indexed-update-operators) that returns an updated copy:
For updating individual elements, JAX provides an [indexed update syntax](https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.ndarray.at.html#jax-numpy-ndarray-at) that returns an updated copy:
```{code-cell} ipython3
:id: 8zqPEAeP3UK5