jax.tree_util.treedef_is_leaf

jax.tree_util.treedef_is_leaf(treedef)[source]