jax.tree_util.treedef_is_leaf

Warning

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

jax.tree_util.treedef_is_leaf¶

jax.tree_util.treedef_is_leaf(treedef)[source]¶