code: def is_array(element: Any) -> bool:
      """Returns `True` if `element` is a JAX array or NumPy array."""
      return isinstance(element, (np.ndarray, np.generic, 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 1 for calling a global function with example usage: is_array(element) and returns: isinstance