code: def field(pytree_node: bool = True, **kwargs):
      metadata = {"pytree_node": pytree_node}
      if "init" in kwargs:
          metadata["init"] = kwargs["init"]
      return dataclasses.field(metadata=metadata, **kwargs)
  
imports: import dataclasses
  import functools
  from typing import Any
  import jax
label_desc: Use image node_id 2 for calling a global function with example usage: field(pytree_node) and returns: dataclasses