code: def celu(
      x: JaxArray,
      /,
      *,
      alpha: float = 1.0,
      complex_mode="jax",
      out: Optional[JaxArray] = None,
  ) -> JaxArray:
      return jax.nn.celu(x, alpha=alpha)
  
imports: from typing import Optional, Union, Literal
  import jax
  import jax.numpy
  from ivy.functional.backends.jax import JaxArray
  from jax import lax
  import ivy
  from ivy.func_wrapper import with_unsupported_dtypes
  from .None import backend_version
label_desc: Use image node_id 8 for calling a global function with example usage: celu() and returns: jax