code: def _nondifferentiable_backward_batch(
      x, batch_axes, *, msg, symbolic
  ):
      (x,) = x
      (batch_axes,) = batch_axes
      return (
          nondifferentiable_backward(x, msg=msg, symbolic=symbolic),
          batch_axes,
      )
  
file_path: equinox/equinox/internal/_nontraceable.py
imports: import functools
  from typing import Optional
  import jax
  import jax.core
  import jax.interpreters.ad
  import jax.interpreters.batching
  import jax.interpreters.mlir
  import jax.tree_util
  from jaxtyping import PyTree
  from .._ad import nondifferentiable
  from .._errors import error_if
  from .._filters import combine, is_array, partition
label_desc: Use image node_id 3 for calling a global function with example usage: _nondifferentiable_backward_batch(x, batch_axes) and returns: nondifferentiable_backward,  batch_axes