code: def _batching(p, b, *, stack, name, intermediates, announce):
      stack = stack + ("vmap",)
      if intermediates:
          announce(name + ":".join(stack))
      out = announce_jaxpr_p.bind(
          *p,
          stack=stack,
          name=name,
          intermediates=intermediates,
          announce=announce,
      )
      return out, b
  
file_path: equinox/equinox/debug/_announce_transform.py
imports: from collections.abc import Callable
  from typing import Any
  import jax
  import jax.core
  import jax.interpreters.ad
  import jax.interpreters.batching
  import jax.interpreters.mlir
  import jax.tree_util
  from .._filters import combine, is_array, partition
label_desc: Use image node_id 6 for calling a global function with example usage: _batching(p, b) and returns: out,  b