code: def roll(
      x: Array,
      /,
      shift: int | tuple[int],
      *,
      axis: int | tuple[int, ...] | None = None,
  ) -> Array:
      """Rolls array elements along a specified axis."""
      return jax.numpy.roll(x, shift=shift, 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 8 for calling a global function with example usage: roll(shift) and returns: jax