jax.tree_util.treedef_is_leaf# jax.tree_util.treedef_is_leaf(treedef)[source]# Parameters: treedef (PyTreeDef) Return type: bool