class_docstr: Base Megatron module inhertied by all Models.
  
  Megatron specific extensions of torch Module with support
  for pipelining
  
  Args:
      config (TransformerConfig): Transformer config
code: def __init__(self, config: TransformerConfig):
      super().__init__()
      self.config = config
  
imports: import torch
  from torch.autograd import Variable
  from torch.nn.parameter import Parameter
  from megatron.core import parallel_state
  from megatron.core.transformer.transformer_config import TransformerConfig
label_desc: Use image node_id 1 to create a new MegatronModule object from inherited base classes: torch.nn with example: obj = MegatronModule(config)