code: def logsumexp(
      a: ArrayLike,
      axis: Axis = None,
      b: ArrayLike | None = None,
      keepdims: bool = False,
      return_sign: bool = False,
  ) -> Array | tuple[Array, Array]:
      ...
  
file_path: jax/jax/_src/ops/special.py
imports: from __future__ import annotations
  from typing import overload, Literal
  import jax
  from jax import lax
  from jax import numpy
  from jax._src.numpy.reductions import _reduction_dims, Axis
  from jax._src.numpy.util import promote_args_inexact
  from jax._src.typing import Array, ArrayLike
  import numpy
label_desc: Use image node_id 3 for calling a global function with example usage: logsumexp(a, axis, b, keepdims, return_sign) without return types