code: def nonzero(
      x: JaxArray,
      /,
      *,
      as_tuple: bool = True,
      size: Optional[int] = None,
      fill_value: Number = 0,
  ) -> Union[JaxArray, Tuple[JaxArray]]:
      res = jnp.nonzero(x, size=size, fill_value=fill_value)
  
      if as_tuple:
          return tuple(res)
  
      return jnp.stack(res, axis=1)
  
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 3 for calling a global function with example usage: nonzero() and returns: jnp, tuple