code: def leaky_relu(
      x: JaxArray,
      /,
      *,
      alpha: float = 0.2,
      complex_mode="jax",
      out: Optional[JaxArray] = None,
  ) -> JaxArray:
      return jnp.asarray(
          jnp.where(x > 0, x, jnp.multiply(x, alpha)), x.dtype
      )
  
file_path: ivy/ivy/functional/backends/jax/activations.py
imports: import jax
  import jax.numpy
  from typing import Optional, Union, Literal
  from ivy.functional.backends.jax import JaxArray
label_desc: Use image node_id 2 for calling a global function with example usage: leaky_relu() and returns: jnp