code: def compute_max_param_mem(region_list: List[Region]) -> float:
      return max(region.param_size for region in region_list)
  
file_path: ColossalAI/colossalai/auto_parallel/offload/util.py
imports: from dataclasses import dataclass
  from typing import List
  import torch
  from colossalai.context.singleton_meta import SingletonMeta
  from colossalai.fx.profiler import calculate_fwd_out, calculate_fwd_tmp
  from .region import Region
label_desc: Use image node_id 2 for calling a global function with example usage: compute_max_param_mem(region_list) and returns: max