jax.tree_util.treedef_children

jax.tree_util.treedef_children(treedef)[source]