.. _api_layers:

Layers
======

Reusable low-level building blocks composed by the DiT encoder, decoder, and
backbone.

.. autoclass:: stix.nn.layers.SinusoidalFourierFeatures

.. autoclass:: stix.nn.layers.AdaptiveLayerNorm

.. autoclass:: stix.nn.layers.SwiGLUFeedForward

.. autoclass:: stix.nn.layers.MultiHeadSelfAttention

.. autoclass:: stix.nn.layers.TransformerBlock

.. autoclass:: stix.nn.layers.RegressionHead