code: def preferred_dtype(config):
      platform = jax.local_devices()[0].platform
      if config.use_mixed_precision:
          if platform == "tpu":
              return jnp.bfloat16
          elif platform == "gpu":
              return jnp.float16
      return jnp.float32
  
imports: import collections
  import functools
  import os
  from absl import logging
  from clu import metric_writers
  from clu import periodic_actions
  from flax import jax_utils
  from flax import linen
  from flax.training import checkpoints
  from flax.training import common_utils
  from flax.training import dynamic_scale
  from flax.training import train_state
  import jax
  import jax.numpy
  import ml_collections
  import numpy
  import optax
  import tensorflow
  import bleu
  import decode
  import input_pipeline
  import models
label_desc: Use image node_id 15 for calling a global function with example usage: preferred_dtype(config) and returns: jnp, jnp, jnp