.. _blocks:

Blocks
======

.. module:: mlip.models.blocks

    .. autoclass:: SpeciesAssignmentBlock

        .. automethod:: __call__

    .. autoclass:: AtomicEnergiesBlock

        .. automethod:: __call__

    .. autoclass:: RadialEmbeddingBlock

        .. automethod:: __call__

    .. autoclass:: SphericalHarmonicsBlock

        .. automethod:: __call__

    .. autoclass:: NodeEmbeddingBlock

        .. automethod:: __call__

    .. autoclass:: MaskPaddedNodeOutputsBlock

        .. automethod:: __call__

    .. autoclass:: MLP

        .. automethod:: __call__