code: def get_layer_ckpt_name_for_rank(base_folder, layer_id, tp_rank):
      ckpt_file = f"{layer_id}-model_{tp_rank:02d}{MODEL_FILE_SUFFIX}"
      ckpt_path = os.path.join(base_folder, ckpt_file)
      return ckpt_path
  
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 3 for calling a global function with example usage: get_layer_ckpt_name_for_rank(base_folder, layer_id, tp_rank) and returns: ckpt_path