code: def unravel_index(
      indices: JaxArray,
      shape: Tuple[int],
      /,
      *,
      out: Optional[JaxArray] = None,
  ) -> Tuple[JaxArray]:
      return jnp.unravel_index(indices.astype(jnp.int32), shape)
  
label_desc: Use image node_id 1 for calling a global function with example usage: unravel_index() and returns: jnp