code: def is_axis_metadata(val: Any) -> bool:
      """Returns whether the argument is an instance of AxisMetadata."""
      return isinstance(val, AxisMetadata)
  
imports: import abc
  import functools
  from typing import Any, Callable, Dict, Generic, Optional, Tuple, TypeVar, Union
  import jax
  from jax.experimental import maps
  from flax import errors, struct
label_desc: Use image node_id 1 for calling a global function with example usage: is_axis_metadata(val) and returns: isinstance