========
GNN Core
========

.. currentmodule:: energnn.model

.. autoclass:: GNN
   :no-members:
   :show-inheritance:

.. autosummary::
   :toctree: _autosummary
   :nosignatures:

   GNN.__call__
   GNN.forward_batch