code: def celu(x: ArrayLike, alpha: ArrayLike = 1.0) -> Array:
      r"""Continuously-differentiable exponential linear unit activation.
  
    Computes the element-wise function:
  
    .. math::
      \mathrm{celu}(x) = \begin{cases}
        x, & x > 0\\
        \alpha \left(\exp(\frac{x}{\alpha}) - 1\right), & x \le 0
      \end{cases}
  
    For more information, see
    `Continuously Differentiable Exponential Linear Units
    <https://arxiv.org/pdf/1704.07483.pdf>`_.
  
    Args:
      x : input array
      alpha : array or scalar (default: 1.0)
  
    Returns:
      An array.
    """
      return jnp.maximum(x, 0.0) + alpha * jnp.expm1(
          jnp.minimum(x, 0.0) / alpha
      )
  
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 11 for calling a global function with example usage: celu(x, alpha) and returns: unknown