code: def scaled_init_method_normal(sigma, num_layers):
      """Init method based on N(0, sigma/sqrt(2*num_layers)."""
      std = sigma / math.sqrt(2.0 * num_layers)
  
      def init_(tensor):
          return torch.nn.init.normal_(tensor, mean=0.0, std=std)
  
      return init_
  
file_path: megatron-lm/megatron/model/utils.py
imports: import math
  import torch
  from megatron import get_args
  from megatron.model import LayerNorm, RMSNorm
label_desc: Use image node_id 2 for calling a global function with example usage: scaled_init_method_normal(sigma, num_layers) and returns: init_, torch