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

Fix cross-reference in pytrees.md

Browse Source
This commit is contained in:
Jake VanderPlas 2021-04-05 16:09:04 -07:00
parent 05e4a99d16
commit 6eb761576c
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
docs/pytrees.md
View File

@ -45,7 +45,7 @@ Example pytrees:
```
JAX can be extended to consider other container types as pytrees; see
[Extending pytrees](extending pytrees) below.
{ref}`extending-pytrees` below.
## Pytrees and JAX functions
@ -187,6 +187,8 @@ for structured in example_containers:
show_example(structured)
```
(extending-pytrees)=
### Extending pytrees
By default, any part of a structured value that is not recognized as an