jax.tree_util.treedef_tuple

Warning

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

jax.tree_util.treedef_tuple¶

jax.tree_util.treedef_tuple(treedefs)[source]¶

Makes a tuple treedef from a list of child treedefs.

previous

jax.tree_util.treedef_is_leaf

next

jax.lib package