code: def normal(
      shape: Tuple[int, ...],
      *,
      mean: float = 0,
      stddev: float = 1,
      generator: Generator = DEFAULT_GENERATOR,
  ):
      """Returns a ``JaxArray`` of shape ``shape`` with random numbers from a normal distribution
      with mean ``mean`` and standard deviation ``stddev``.
  
      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.normal(generator(), shape=shape) * stddev + mean
  
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 1 for calling a global function with example usage: normal(shape) and returns: unknown