code: def rng():  # Can't be a constant, because JAX has to init itself in main first.
      return fastmath.random.get_prng(0)
  
file_path: trax/trax/layers/initializers_test.py
imports: from absl.testing import absltest
  import numpy
  from trax import fastmath
  from trax import test_utils
  import trax.layers
label_desc: Use image node_id 1 for calling a global function with example usage: rng() and returns: fastmath