JaxHyperEdgeSet.tree_flatten¶ JaxHyperEdgeSet.tree_flatten()[source]¶ Flattens a PyTree, required for JAX compatibility. :returns: a tuple of values and keys Return type: tuple