code: def _state_unflatten(
      static: tp.Tuple[Path, ...] | None,
      leaves: tp.Tuple[Leaf, ...] | tuple[dict[str, Leaf]],
  ):
      return State(zip(static, leaves)) if static else State(leaves[0])
  
imports: from __future__ import annotations
  import typing
  import jax
  import jax.tree_util
  from flax import traverse_util
  from flax.experimental.nnx.nnx import filterlib, reprlib
  from flax.experimental.nnx.nnx.variables import Variable
label_desc: Use image node_id 2 for calling a global function with example usage: _state_unflatten(static, leaves) and returns: unknown