code: def _build_pp_transformer_map(self):
      data_map = {}
      transformer_layers = self.layer_keys[1:-1]
      layers_per_pp = len(transformer_layers) // self.pp_degree
      data_map = {
          i: transformer_layers[
              i * layers_per_pp : (i + 1) * layers_per_pp
          ]
          for i in range(0, self.pp_degree)
      }
      return data_map
  
imports: import os
  from typing import Dict
  import torch
  from .reshape_3d_utils import model_3d_desc
  from .reshape_utils import basic_folder_validation, merge_state, partition_data, get_files, get_files_with_prefix
  from .constants import MODEL_FILE_PREFIX, LAYER_FILE_PREFIX
  from .reshape_meg_2d import reshape_meg_2d_parallel, meg_2d_parallel_map
  from .zero_checkpoint import ZeROCheckpoint
  from .constants import *
label_desc: Use image node_id 28 for calling the DeepSpeedCheckpoint obj's underlying member method code with example usage: obj._build_pp_transformer_map() and returns: data_map