code: def jax2tfjs(harness: ModelHarness):
      """Converts the given `test_case` using the TFjs converter."""
      with tempfile.TemporaryDirectory() as model_dir:
          tfjs.converters.convert_jax(
              apply_fn=harness.apply,
              params=harness.variables,
              input_signatures=harness.tf_input_signature,
              polymorphic_shapes=harness.polymorphic_shapes,
              model_dir=model_dir,
          )
  
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 2 for calling a global function with example usage: jax2tfjs(harness) without return types