include jax_conversion target to fix nightly test (#6767)

Author: pyu10055

tfjs-converter/python/setup.py

@@ -91,6 +91,7 @@ def get_extra_requires(path, add_all=True):
         'tensorflowjs.converters.graph_rewrite_util',
         'tensorflowjs.converters.keras_h5_conversion',
         'tensorflowjs.converters.keras_tfjs_loader',
+        'tensorflowjs.converters.jax_conversion',
         'tensorflowjs.converters.tf_saved_model_conversion_v2',
         'tensorflowjs.converters.wizard',
     ],

tfjs-converter/python/tensorflowjs/converters/BUILD

@@ -12,6 +12,7 @@ py_library(
     visibility = ["//visibility:public"],
     deps = [
         ":converter",
+        ":jax_conversion",
         ":keras_h5_conversion",
         ":keras_tfjs_loader",
         ":tf_saved_model_conversion_v2",