code: def cross(x1, x2, /, *, axis=-1):
      """
      Returns the cross product of 3-element vectors.
      """
      return jax.numpy.linalg.cross(x1, x2, axis=axis)
  
imports: from functools import partial
  from typing import NamedTuple
  import jax
  from jax.experimental.array_api._data_type_functions import _promote_to_default_dtype
label_desc: Use image node_id 2 for calling a global function with example usage: cross() and returns: jax