code: def is_array_like(element: Any) -> bool:
      """Returns `True` if `element` is a JAX array, a NumPy array, or a Python
      `float`/`complex`/`bool`/`int`.
      """
      return isinstance(
          element,
          (
              jax.Array,
              np.ndarray,
              np.generic,
              float,
              complex,
              bool,
              int,
          ),
      ) or hasattr(element, "__jax_array__")
  
file_path: equinox/equinox/_filters.py
imports: from collections.abc import Callable
  from typing import Any, Optional, Union
  import jax
  import jax.numpy
  import jax.tree_util
  import numpy
  from jaxtyping import PyTree
label_desc: Use image node_id 2 for calling a global function with example usage: is_array_like(element) and returns: unknown