code: def main(args):
      collect_profile(
          args.port,
          args.duration_in_ms,
          args.host,
          args.log_dir,
          args.host_tracer_level,
          args.device_tracer_level,
          args.python_tracer_level,
          args.no_perfetto_link,
      )
  
file_path: jax/jax/collect_profile.py
imports: from __future__ import annotations
  import argparse
  import gzip
  import os
  import pathlib
  import tempfile
  from jax._src import profiler
label_desc: Use image node_id 2 for calling a global function with example usage: main(args) without return types