JaxGraphShape.tree_flatten
==========================

.. currentmodule:: energnn.graph

.. automethod:: JaxGraphShape.tree_flatten