code: def round(x):
      return np.trunc(
          x
          + np.copysign(
              np.nextafter(
                  np.array(0.5, dtype=x.dtype),
                  np.array(0.0, dtype=x.dtype),
                  dtype=x.dtype,
              ),
              x,
          )
      ).astype(x.dtype)
  
imports: import builtins
  import collections
  import itertools
  import numpy
  import opt_einsum
  import scipy.special
  from jax._src import dtypes
  from jax._src import util
label_desc: Use image node_id 1 for calling a global function with example usage: round(x) and returns: np