code: def safe_get_local_fp32_param(param):
      """Get the fp32 partitioned parameter.
      Args:
          param (``torch.nn.Parameter``): A model parameter
      """
      # ZeRO stage 3 param
      if hasattr(param, "ds_id"):
          return param._z3_optimizer.get_local_fp32_param(param)
  
      return None
  
imports: import torch
  from dataclasses import dataclass
  from deepspeed import comm
  from typing import Dict
label_desc: Use image node_id 10 for calling a global function with example usage: safe_get_local_fp32_param(param) and returns: None, param