code: def zero3_partitioned_param_info(unpartitioned_numel, world_size):
      remainder = unpartitioned_numel % world_size
      padding_numel = (world_size - remainder) if remainder else 0
      partitioned_numel = math.ceil(unpartitioned_numel / world_size)
      return partitioned_numel, padding_numel
  
imports: import argparse
  import torch
  import glob
  import math
  import os
  import re
  from collections import OrderedDict
  from dataclasses import dataclass
  from deepspeed.utils import logger
  from deepspeed.checkpoint.constants import DS_VERSION, OPTIMIZER_STATE_DICT, SINGLE_PARTITION_OF_FP32_GROUPS, FP32_FLAT_GROUPS, ZERO_STAGE, PARTITION_COUNT, PARAM_SHAPES, BUFFER_NAMES, FROZEN_PARAM_SHAPES, FROZEN_PARAM_FRAGMENTS
label_desc: Use image node_id 13 for calling a global function with example usage: zero3_partitioned_param_info(unpartitioned_numel, world_size) and returns: partitioned_numel,  padding_numel