docstr: Random numbers should be handled correctly.
code: def test_random(self):
      """Random numbers should be handled correctly."""
      n = 100
  
      @transform.transform
      @layer_stack.layer_stack(n)
      def add_random(x):
          x = x + jax.random.normal(base.next_rng_key())
          return x
  
      # Evaluate a bunch of times
      key, *keys = jax.random.split(jax.random.PRNGKey(7), 1024 + 1)
      params = add_random.init(key, 0.0)
      apply_fn = jax.jit(add_random.apply)
      values = [apply_fn(params, key, 0.0) for key in keys]
  
      # Should be roughly N(0, sqrt(n))
      cdf = stats.norm(scale=np.sqrt(n)).cdf
      _, p = stats.kstest(
          values, cdf
      )  # pytype: disable=attribute-error
      self.assertLess(p, 0.1)
  
imports: import functools
  import re
  from typing import Optional
  from absl.testing import absltest
  from absl.testing import parameterized
  from haiku._src import base
  from haiku._src import basic
  from haiku._src import config
  from haiku._src import initializers
  from haiku._src import layer_stack
  from haiku._src import module
  from haiku._src import multi_transform
  from haiku._src import transform
  from haiku._src import utils
  import jax
  import jax.numpy
  import numpy
  from scipy import stats
label_desc: Use image node_id 8 for calling the LayerStackTest obj's underlying member method code with example usage: obj.test_random() and returns: x