code: def load_json(file_path: Union[str, os.PathLike]) -> Dict[str, Any]:
      """
      Load file in JSON format
      """
      with open(file=file_path, mode="r", encoding="utf-8") as fp:
          return json.load(fp)
  
file_path: ColossalAI/applications/Colossal-LLaMA-2/colossal_llama2/utils/ckpt_io.py
imports: import json
  import os
  from typing import Any, Dict, Tuple, Union
  import torch
  from torch.optim.optimizer import Optimizer
  from torch.optim.lr_scheduler import _LRScheduler
  from colossalai.booster import Booster
  from colossalai.cluster import DistCoordinator
label_desc: Use image node_id 1 for calling a global function with example usage: load_json(file_path) and returns: json