Toggle navigation

.rst .pdf

Warning

This page was created from a pull request (#9655).

jax.tree_util.build_tree

jax.tree_util.build_tree(treedef, xs)

previous

jax.tree_util.all_leaves

next

jax.tree_util.register_pytree_node