code: def where(
      condition: JaxArray,
      x1: JaxArray,
      x2: JaxArray,
      /,
      *,
      out: Optional[JaxArray] = None,
  ) -> JaxArray:
      x1, x2 = ivy.promote_types_of_inputs(x1, x2)
      return ivy.astype(
          jnp.where(condition, x1, x2), x1.dtype, copy=False
      )
  
file_path: ivy/ivy/functional/backends/jax/searching.py
imports: from numbers import Number
  from typing import Optional, Tuple, Union
  import jax.numpy
  import ivy
  from ivy.functional.backends.jax import JaxArray
  from .None import backend_version
  from ivy.func_wrapper import with_unsupported_dtypes
label_desc: Use image node_id 4 for calling a global function with example usage: where() and returns: ivy