code: def leaky_relu(
      x: ArrayLike, negative_slope: ArrayLike = 1e-2
  ) -> Array:
      r"""Leaky rectified linear unit activation function.
  
    Computes the element-wise function:
  
    .. math::
      \mathrm{leaky\_relu}(x) = \begin{cases}
        x, & x \ge 0\\
        \alpha x, & x < 0
      \end{cases}
  
    where :math:`\alpha` = :code:`negative_slope`.
  
    Args:
      x : input array
      negative_slope : array or scalar specifying the negative slope (default: 0.01)
  
    Returns:
      An array.
  
    See also:
      :func:`relu`
    """
      numpy_util.check_arraylike("leaky_relu", x)
      x_arr = jnp.asarray(x)
      return jnp.where(x_arr >= 0, x_arr, negative_slope * x_arr)
  
imports: from __future__ import annotations
  from functools import partial
  import operator
  import numpy
  from typing import Any
  import jax
  import jax.numpy
  from jax import custom_jvp
  from jax import lax
  from jax._src import config
  from jax._src import core
  from jax._src import dtypes
  from jax._src import util
  from jax._src.core import AxisName
  from jax._src.numpy import util
  from jax._src.typing import Array, ArrayLike
  from jax._src.ops.special import logsumexp
label_desc: Use image node_id 9 for calling a global function with example usage: leaky_relu(x, negative_slope) and returns: jnp