code: def KaimingNormalInitializer(out_dim=-1, in_dim=-2, param=0.0):
      """Returns an initializer for random Kaiming-scaled coefficients."""
      return ScaledInitializer(
          out_dim,
          in_dim,
          2.0 / jnp.sqrt(1 + param**2),
          "fan_in",
          "normal",
      )
  
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 11 for calling a global function with example usage: KaimingNormalInitializer(out_dim, in_dim, param) and returns: ScaledInitializer