code: def expand_dims(x: Array, /, *, axis: int = 0) -> Array:
      """Expands the shape of an array by inserting a new axis (dimension) of size one at the position specified by axis."""
      if axis < -x.ndim - 1 or axis > x.ndim:
          raise IndexError(
              f"{axis=} is out of bounds for array of dimension {x.ndim}"
          )
      return jax.numpy.expand_dims(x, axis=axis)
  
file_path: jax/jax/experimental/array_api/_manipulation_functions.py
imports: from __future__ import annotations
  import jax
  from jax import Array
  from jax.experimental.array_api._data_type_functions import result_type
label_desc: Use image node_id 4 for calling a global function with example usage: expand_dims() and returns: jax