optax.tree_utils.tree_max# optax.tree_utils.tree_max( tree: Any, ) → jax.typing.ArrayLike[source]# Compute the max of all the elements in a pytree. Parameters: tree – pytree. Returns: a scalar value.
