code: def _split_in_out_axes(xs: Mapping[CollectionFilter, Any]):
      unpack = lambda v: v.axis if isinstance(v, (In, Out)) else v
      in_axes = {
          k: unpack(v) for k, v in xs.items() if not isinstance(v, Out)
      }
      out_axes = {
          k: unpack(v) for k, v in xs.items() if not isinstance(v, In)
      }
      return in_axes, out_axes
  
imports: import collections
  import dataclasses
  import functools
  from typing import Any, Callable, Dict, Generic, Iterable, List, Mapping, Optional, Sequence, Tuple, TypeVar, Union
  import warnings
  from flax import traceback_util
  import jax
  from jax import random
  from .None import axes_scan, meta
  from .frozen_dict import freeze, unfreeze
  from .scope import CollectionFilter, DenyList, Filter, PRNGSequenceFilter, Scope, group_collections, in_filter, intersect_filters, is_filter_empty, subtract_filters, union_filters
label_desc: Use image node_id 9 for calling a global function with example usage: _split_in_out_axes(xs) and returns: in_axes,  out_axes