code: def argsort(
      x: Array,
      /,
      *,
      axis: int = -1,
      descending: bool = False,
      stable: bool = True,
  ) -> Array:
      """Returns the indices that sort an array x along a specified axis."""
      return jax.numpy.argsort(
          x, axis=axis, descending=descending, stable=stable
      )
  
label_desc: Use image node_id 1 for calling a global function with example usage: argsort() and returns: jax