Models
===================================

Base Class
----------
.. automodule:: lagrangebench.models.base
    :members:

GNS
---
.. automodule:: lagrangebench.models.gns
    :members:

SEGNN
-----
.. automodule:: lagrangebench.models.segnn
    :members:

EGNN
----
.. automodule:: lagrangebench.models.egnn
    :members:

PaiNN
-----
.. automodule:: lagrangebench.models.painn
    :members:

Linear
------
.. automodule:: lagrangebench.models.linear
    :members:

Utils
-----
.. automodule:: lagrangebench.models.utils
    :members:
    :exclude-members: __getnewargs__, __new__, __repr__
