code: def zeros(
      shape: Union[ivy.NativeShape, Sequence[int]],
      *,
      dtype: jnp.dtype,
      device: jaxlib.xla_extension.Device = None,
      out: Optional[JaxArray] = None,
  ) -> JaxArray:
      return jnp.zeros(shape, dtype)
  
imports: from numbers import Number
  import numpy
  from typing import Union, Optional, List, Sequence, Tuple
  import jax.dlpack
  import jax.numpy
  import jax._src
  import jaxlib.xla_extension
  import ivy
  from ivy import as_native_dtype
  from ivy.functional.backends.jax import JaxArray
  from ivy.functional.backends.jax.device import dev
  from ivy.functional.ivy.creation import _asarray_to_native_arrays_and_back, _asarray_infer_device, _asarray_infer_dtype, _asarray_handle_nestable, NestedSequence, SupportsBufferProtocol, _asarray_inputs_to_native_shapes
label_desc: Use image node_id 16 for calling a global function with example usage: zeros(shape) and returns: jnp