code: def run(main):
      # Provide access to --jax_backend_target and --jax_xla_backend flags.
      jax.config.config_with_absl()
      app.run(functools.partial(_run_main, main=main))
  
file_path: scenic/scenic/app.py
imports: import functools
  from absl import app
  from absl import flags
  from absl import logging
  from clu import metric_writers
  from clu import platform
  import flax
  import flax.linen
  import jax
  from ml_collections import config_flags
  import tensorflow
label_desc: Use image node_id 1 for calling a global function with example usage: run(main) without return types