code: def is_multi_device_array(value: Any) -> bool:
      """Instruct Orbax to save this array with Tensorstore instead of msgpack."""
      if isinstance(value, jax.Array):
          return not value.is_fully_replicated
      return False
  
file_path: flax/flax/training/orbax_utils.py
imports: import dataclasses
  import inspect
  import warnings
  from typing import Any, Optional
  import jax
  import numpy
  import orbax.checkpoint
  from jax.sharding import Mesh
label_desc: Use image node_id 1 for calling a global function with example usage: is_multi_device_array(value) and returns: False, unknown