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

DOC: fix doc formatting

Browse Source
This commit is contained in:
Jake VanderPlas 2023-06-01 03:37:21 -07:00
parent 4ffeafed85
commit e5cd69479b
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: