code: def _wrap_between(x, _a):
      """Wraps `x` between `[-a, a]`."""
      a = _constant_like(x, _a)
      two_a = _constant_like(x, 2 * _a)
      zero = _constant_like(x, 0)
      rem = lax.rem(lax.add(x, a), two_a)
      rem = lax.select(lax.lt(rem, zero), lax.add(rem, two_a), rem)
      return lax.sub(rem, a)
  
imports: from __future__ import annotations
  from functools import partial
  import operator
  from textwrap import dedent
  from typing import Any, Callable, overload
  import numpy
  from jax._src import core
  from jax._src import dtypes
  from jax._src.api import jit
  from jax._src.custom_derivatives import custom_jvp
  from jax._src.lax import lax
  from jax._src.typing import Array, ArrayLike
  from jax._src.numpy.util import check_arraylike, promote_args, promote_args_inexact, promote_args_numeric, promote_dtypes_inexact, promote_dtypes_numeric, promote_shapes, _where, _wraps, check_no_float0s
label_desc: Use image node_id 27 for calling a global function with example usage: _wrap_between(x, _a) and returns: lax