PytreeEdge# class brainstate.graph.PytreeEdge(metadata, fields)[source]# An inline pytree container. Parameters: metadata (Any) – The JAX PyTreeDef (treedef) of one flattening level of the container, used to rebuild it on decode. fields (tuple) – The ordered child edges keyed by their per-level path key.
