code: def _unvmap_all_batch(x, batch_axes):
      (x,) = x
      return unvmap_all(x), batching.not_mapped
  
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 4 for calling a global function with example usage: _unvmap_all_batch(x, batch_axes) and returns: unvmap_all,  batching