code: def randint(
      shape: Tuple[int, ...],
      low: int,
      high: int,
      generator: Generator = DEFAULT_GENERATOR,
  ):
      """Returns a ``JaxAarray`` of shape ``shape`` with random integers in {low, ..., high-1}.
  
      NOTE: if random numbers are generated inside a jitted, parallelized or vectorized function
      then generator variables (including DEFAULT_GENERATOR) have to be added to the
      variable collection."""
      return jr.randint(
          generator(), shape=shape, minval=low, maxval=high
      )
  
file_path: objax/objax/random/random.py
imports: from typing import Optional, Tuple
  import jax.random
  from objax.module import Module
  from objax.util import class_name
  from objax.variable import RandomState, VarCollection
label_desc: Use image node_id 2 for calling a global function with example usage: randint(shape, low, high, generator) and returns: jr