code: def _is_array_like_internal(x):
      assert type(x) is not ad.UndefinedPrimal
      # Not `type(x) in (...)` as that doesn't handle stuff like ConcreteArrays.
      return is_array(x) or isinstance(
          x, (jax.core.ShapedArray, _WrappedPrimal)
      )
  
imports: import functools
  from typing import Any
  import jax
  import jax.core
  import jax.interpreters.ad
  import jax.interpreters.batching
  import jax.interpreters.mlir
  import jax.tree_util
  from jaxtyping import PyTree
  from .._filters import combine, is_array, is_array_like, partition
  from .._tree import tree_equal
label_desc: Use image node_id 3 for calling a global function with example usage: _is_array_like_internal(x) and returns: unknown