From 008c25a36977c59b0446816b2151e0a7fe86b209 Mon Sep 17 00:00:00 2001 From: mikcl Date: Thu, 26 Dec 2024 01:12:05 +0000 Subject: [PATCH] Fix formatting in the docs for transposing pytrees --- docs/working-with-pytrees.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/working-with-pytrees.md b/docs/working-with-pytrees.md index 537a4df3e5a6..ffa47eba07c0 100644 --- 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: