code: def __init__(self, torch_pg: ProcessGroup):
      super().__init__(torch_pg)
  
      # record the padding size of each param
      self._padding_map = dict()
  
      # mapping working param and master param
      self.master_to_working_param = dict()
      self.working_to_master_param = dict()
  
file_path: ColossalAI/colossalai/zero/low_level/bookkeeping/parameter_store.py
imports: from torch import Tensor
  from torch.distributed import ProcessGroup
  from .base_store import BaseStore
label_desc: Use image node_id 1 to create a new ParameterStore object from inherited base classes: BaseStore with example: obj = ParameterStore(torch_pg)