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

Merge pull request #16212 from jakevdp:fix-doc

Browse Source 
PiperOrigin-RevId: 536977403
This commit is contained in:
jax authors 2023-06-01 03:59:29 -07:00
commit eb41e9c331
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/jax-101/05.1-pytrees.ipynb
View File

@ -346,7 +346,7 @@
"\n",
"* [`jax.tree_util.tree_map_with_path`](https://jax.readthedocs.io/en/latest/_autosummary/jax.tree_util.tree_map_with_path.html): Works similarly with `jax.tree_util.tree_map`, but the function also takes key paths as arguments.\n",
"\n",
"* [`jax.tree_util.keystr`]((https://jax.readthedocs.io/en/latest/_autosummary/jax.tree_util.keystr.html)): Given a general key path, returns a reader-friendly string expression.\n",
"* [`jax.tree_util.keystr`](https://jax.readthedocs.io/en/latest/_autosummary/jax.tree_util.keystr.html): Given a general key path, returns a reader-friendly string expression.\n",
"\n",
"One use case is to print debugging information related to a certain leaf value:"
]

2
docs/jax-101/05.1-pytrees.md
View File

@ -199,7 +199,7 @@ The APIs for working with key paths are:
* [`jax.tree_util.tree_map_with_path`](https://jax.readthedocs.io/en/latest/_autosummary/jax.tree_util.tree_map_with_path.html): Works similarly with `jax.tree_util.tree_map`, but the function also takes key paths as arguments.
* [`jax.tree_util.keystr`]((https://jax.readthedocs.io/en/latest/_autosummary/jax.tree_util.keystr.html)): Given a general key path, returns a reader-friendly string expression.
* [`jax.tree_util.keystr`](https://jax.readthedocs.io/en/latest/_autosummary/jax.tree_util.keystr.html): Given a general key path, returns a reader-friendly string expression.
One use case is to print debugging information related to a certain leaf value: