code: def jax2tf_convert(harness: ModelHarness, enable_xla: bool = True):
      return jax2tf.convert(
          harness.apply_with_vars,
          enable_xla=enable_xla,
          polymorphic_shapes=harness.polymorphic_shapes,
      )
  
file_path: jax/jax/experimental/jax2tf/tests/converters.py
imports: import dataclasses
  import functools
  import tempfile
  from typing import Any, Callable
  from jax.experimental import jax2tf
  import tensorflow
  import tensorflowjs
  from jax.experimental.jax2tf.tests.model_harness import ModelHarness
label_desc: Use image node_id 1 for calling a global function with example usage: jax2tf_convert(harness, enable_xla) and returns: jax2tf