JaxHyperEdgeSet.tree_flatten
============================

.. currentmodule:: energnn.graph

.. automethod:: JaxHyperEdgeSet.tree_flatten