code: def _uniform_jitter(mean_num_leapfrog_steps, step, seed):
      del step
      return util.random_integer(
          [],
          dtype=jnp.int32,
          minval=1,
          maxval=2 * mean_num_leapfrog_steps,
          seed=seed,
      )
  
imports: import functools
  import typing
  from typing import Any, Callable, Iterable, NamedTuple, Optional
  import numpy
  from fun_mc import backend
  from fun_mc import fun_mc_lib
  from fun_mc import malt
  from fun_mc import sga_hmc
  from fun_mc.malt import *
  from fun_mc.sga_hmc import *
label_desc: Use image node_id 5 for calling a global function with example usage: _uniform_jitter(mean_num_leapfrog_steps, step, seed) and returns: util