code: def AtariConvInit(kernel_shape, rng, dtype=jnp.float32):
      """The standard init for Conv laters and Atari."""
      filter_height, filter_width, fan_in, _ = kernel_shape
      std = 1 / jnp.sqrt(fan_in * filter_height * filter_width)
      return random.uniform(
          rng, kernel_shape, dtype, minval=-std, maxval=std
      )
  
file_path: trax/trax/layers/initializers.py
imports: from absl import logging
  import numpy
  import tensorflow.compat.v2
  from trax.fastmath import numpy
  from trax.fastmath import random
label_desc: Use image node_id 14 for calling a global function with example usage: AtariConvInit(kernel_shape, rng, dtype) and returns: random