class_docstr: Base class to apply hidden state transformations
class_docstr_tok: ["Base","class","to","apply","hidden","state","transformations"]
code: def __init__(
      self,
      name: str = "",
      model_parallel_cfg: ModelParallelConfig = None,
  ):
      super().__init__()
  
      self.name = name
      self.model_parallel_cfg = model_parallel_cfg
  
file_path: NeMo/nemo/collections/nlp/modules/common/megatron/hiddens/megatron_hidden_transform.py
imports: import math
  import torch
  from nemo.collections.nlp.modules.common.megatron.utils import ApexGuardDefaults, init_method_normal
label_desc: Use image node_id 1 to create a new MegatronBaseHiddenTransform object from inherited base classes: torch.nn with example: obj = MegatronBaseHiddenTransform(name, model_parallel_cfg)