code: def _round_up(i, n):
      return ((i + n - 1) // n) * n
  
imports: from __future__ import annotations
  from functools import partial
  from typing import NamedTuple
  import jax
  import jax._src.numpy.lax_numpy
  import jax._src.numpy.linalg
  from jax._src.numpy import reductions
  from jax._src.numpy import ufuncs
  from jax import lax
  from jax._src.lax import qdwh
  from jax._src.lax import linalg
  from jax._src.lax.stack import Stack
label_desc: Use image node_id 1 for calling a global function with example usage: _round_up(i, n) and returns: unknown