Merge pull request #25686 from Mikcl:docs/working-with-pytrees-format…

…ting PiperOrigin-RevId: 711448394
jax-ml · Jan 2, 2025 · 800f903 · 800f903
2 parents 726950b + 008c25a
commit 800f903
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/working-with-pytrees.md b/docs/working-with-pytrees.md
@@ -490,7 +490,7 @@ This section covers some of the most common patterns with JAX pytrees.
 
 ### Transposing pytrees with `jax.tree.map` and `jax.tree.transpose`
 
-To transpose a pytree (turn a list of trees into a tree of lists), JAX has two functions: {func} `jax.tree.map` (more basic) and {func}`jax.tree.transpose` (more flexible, complex and verbose).
+To transpose a pytree (turn a list of trees into a tree of lists), JAX has two functions: {func}`jax.tree.map` (more basic) and {func}`jax.tree.transpose` (more flexible, complex and verbose).
 
 **Option 1:** Use {func}`jax.tree.map`. Here's an example: