code: def _normalize_axis_index(ax: int, ndim: int) -> int:
      if ax >= ndim or ax < -ndim:
          raise ValueError("axis index is out of range")
      return (ax + ndim) % ndim
  
imports: import operator
  from typing import Optional, Union, Tuple, List, Sequence
  from numbers import Number
  from ivy import promote_types_of_inputs, default_float_dtype, is_float_dtype
  from ivy.func_wrapper import with_supported_dtypes
  from ivy.functional.backends.jax import JaxArray
  import jax.numpy
  import jax.scipy
  import jax.lax
  from ..None import backend_version
label_desc: Use image node_id 18 for calling a global function with example usage: _normalize_axis_index(ax, ndim) and returns: unknown