class_docstr: RMS LayerNorm
code: def __init__(
      self,
      dim_norm: int,
      dtype: torch.dtype = torch.half,
      eps: float = 1e-6,
      init_var: float = 1.0,
  ):
      super().__init__()
  
      self.eps = eps
      self.dim_norm = dim_norm
      self.weight = bmt.DistributedParameter(
          torch.full((dim_norm,), init_var, dtype=dtype)
      )
  
imports: import torch
  import bmtrain
label_desc: Use image node_id 1 to create a new LayerNorm object from inherited base classes: bmt with example: obj = LayerNorm(dim_norm, dtype, eps, init_var)