code: def _frozen_dict_state_dict(xs):
      return {
          key: serialization.to_state_dict(value)
          for key, value in xs.items()
      }
  
imports: import collections
  from types import MappingProxyType
  from typing import Any, Dict, Hashable, Mapping, Tuple, TypeVar, Union
  import jax
  from flax import serialization
label_desc: Use image node_id 8 for calling a global function with example usage: _frozen_dict_state_dict(xs) and returns: unknown