code: def rint(x: ArrayLike, /) -> Array:
      check_arraylike("rint", x)
      dtype = dtypes.dtype(x)
      if dtype == bool or dtypes.issubdtype(dtype, np.integer):
          return lax.convert_element_type(x, dtypes.float_)
      if dtypes.issubdtype(dtype, np.complexfloating):
          return lax.complex(rint(lax.real(x)), rint(lax.imag(x)))
      return lax.round(x, lax.RoundingMethod.TO_NEAREST_EVEN)
  
imports: from __future__ import annotations
  from functools import partial
  import operator
  from textwrap import dedent
  from typing import Any, Callable, overload
  import numpy
  from jax._src import core
  from jax._src import dtypes
  from jax._src.api import jit
  from jax._src.custom_derivatives import custom_jvp
  from jax._src.lax import lax
  from jax._src.typing import Array, ArrayLike
  from jax._src.numpy.util import check_arraylike, promote_args, promote_args_inexact, promote_args_numeric, promote_dtypes_inexact, promote_dtypes_numeric, promote_shapes, _where, _wraps, check_no_float0s
label_desc: Use image node_id 16 for calling a global function with example usage: rint() and returns: lax, lax, lax