JaxGraph.tree_flatten
=====================

.. currentmodule:: energnn.graph

.. automethod:: JaxGraph.tree_flatten