code: def uniform(
      shape: Tuple[int, ...], generator: Generator = DEFAULT_GENERATOR
  ):
      """Returns a ``JaxArray`` of shape ``shape`` with random numbers from a uniform distribution [0, 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.uniform(generator(), shape=shape)
  
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 4 for calling a global function with example usage: uniform(shape, generator) and returns: jr