code: def triu_indices(
      n_rows: int,
      n_cols: Optional[int] = None,
      k: int = 0,
      /,
      *,
      device: jaxlib.xla_extension.Device = None,
  ) -> Tuple[JaxArray]:
      return jnp.triu_indices(n=n_rows, k=k, m=n_cols)
  
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 21 for calling a global function with example usage: triu_indices() and returns: jnp