code: def log_note(note: str):
      """Write note to XManager and also log to the console."""
      if jax.process_index() == 0:  # Only perform on the lead_host
          logging.info(note)
  
file_path: scenic/scenic/projects/objectvivit/train_utils.py
imports: from typing import Any, Dict, Tuple
  from absl import logging
  import jax
  import jax.numpy
  import numpy
  from scenic.model_lib.base_models import model_utils
label_desc: Use image node_id 5 for calling a global function with example usage: log_note(note) without return types