code: def trilu(
      x: JaxArray,
      /,
      *,
      k: int = 0,
      upper: bool = True,
      out: Optional[JaxArray] = None,
  ) -> JaxArray:
      if upper:
          return jnp.triu(x, k)
      return jnp.tril(x, k)
  
file_path: ivy/ivy/functional/backends/jax/experimental/creation.py
imports: from typing import Optional, Tuple
  import math
  import jax
  import jax.numpy
  import jaxlib.xla_extension
  from ivy.functional.backends.jax import JaxArray
  import ivy
label_desc: Use image node_id 8 for calling a global function with example usage: trilu() and returns: jnp, jnp