code: def parse_args():
      # Parse command line arguments
      parser = argparse.ArgumentParser()
      parser.add_argument(
          "--config",
          type=str,
          default=None,
          help="Config file location",
      )
      parser.add_argument(
          "--ckpt-step",
          type=int,
          default=-1,
          help="Step number of the checkpoint to convert (if not specified, converts the most recent checkpoint)",
      )
      parser.add_argument(
          "--f16",
          default=False,
          action="store_true",
          help="Convert to float16 (instead of bfloat16)",
      )
  
      args = parser.parse_args()
      return args
  
file_path: mesh-transformer-jax/slim_model.py
imports: import argparse
  import json
  import time
  import jax
  import numpy
  import optax
  from mesh_transformer import util
  from mesh_transformer.checkpoint import read_ckpt, write_ckpt
  from mesh_transformer.transformer_shard import CausalTransformer
  from smart_open import open
  from mesh_transformer.util import clip_by_global_norm, to_bf16, to_f16
label_desc: Use image node_id 1 for calling a global function with example usage: parse_args() and returns: args