Skip to content

Commit

Permalink
Fix formatting in docs for transposing pytrees
Browse files Browse the repository at this point in the history
  • Loading branch information
Mikcl committed Dec 26, 2024
1 parent 42a0d55 commit f70dc2d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/working-with-pytrees.md
Original file line number Diff line number Diff line change
Expand Up @@ -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:

Expand Down

0 comments on commit f70dc2d

Please sign in to comment.