code: def is_jax_tensor(x):
      """
      Tests if `x` is a Jax tensor or not. Safe to call even if jax is not installed.
      """
      return False if not is_flax_available() else _is_jax(x)
  
imports: import inspect
  import tempfile
  from collections import OrderedDict, UserDict
  from collections.abc import MutableMapping
  from contextlib import ExitStack, contextmanager
  from dataclasses import fields
  from enum import Enum
  from typing import Any, ContextManager, List, Tuple
  import numpy
  from .import_utils import is_flax_available, is_tf_available, is_torch_available, is_torch_fx_proxy
label_desc: Use image node_id 11 for calling a global function with example usage: is_jax_tensor(x) and returns: unknown