code: def get_zero_ckpt_name_for_rank(base_folder, dp_rank, mp_rank):
      zero_prefix = f"{ZERO_FILE_PREFIX}{dp_rank}"
      mp_rank_string = f"_{MODEL_FILE_PREFIX}{mp_rank:02d}"
      zero_ckpt_name = os.path.join(
          base_folder,
          zero_prefix + mp_rank_string + OPTIM_FILE_SUFFIX,
      )
      return zero_ckpt_name
  
imports: import os
  import torch
  from .constants import MODEL_FILE_PREFIX, MODEL_FILE_SUFFIX, OPTIM_FILE_SUFFIX, ZERO_FILE_PREFIX
label_desc: Use image node_id 2 for calling a global function with example usage: get_zero_ckpt_name_for_rank(base_folder, dp_rank, mp_rank) and returns: zero_ckpt_name