code: def unvmap_max(x: Int[ArrayLike, "..."]) -> Int[Array, ""]:
      """As `jnp.max`, but ignores batch dimensions."""
      return cast(Array, unvmap_max_p.bind(x))
  
file_path: equinox/equinox/_unvmap.py
imports: from typing import cast
  import jax
  import jax.core
  import jax.interpreters.batching
  import jax.interpreters.mlir
  import jax.numpy
  from jaxtyping import Array, ArrayLike, Bool, Int
label_desc: Use image node_id 9 for calling a global function with example usage: unvmap_max(x) and returns: cast