jax.tree_util.tree_leaves_with_path# jax.tree_util.tree_leaves_with_path(tree, is_leaf=None, is_leaf_takes_path=False)[source]# Alias of jax.tree.leaves_with_path(). Parameters: tree (Any) is_leaf (Callable[..., bool] | None) is_leaf_takes_path (bool) Return type: list[tuple[KeyPath, Any]]
