code: def save_json(
      data: Dict[str, Any], file_path: Union[str, os.PathLike]
  ) -> None:
      """
      Save as JSON format
      """
      with open(file=file_path, mode="w", encoding="utf-8") as fp:
          json.dump(data, fp=fp, ensure_ascii=False, indent=4)
  
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 2 for calling a global function with example usage: save_json(data, file_path) without return types