.. _api_dit:

DiT components
==============

The DiT (diffusion transformer) backbone and its matching encoder, decoder, and
time/noise context encoder.

.. autoclass:: stix.nn.DiTEncoder

.. autoclass:: stix.nn.DiTBackbone

.. autoclass:: stix.nn.DiTDecoder

.. autoclass:: stix.nn.TimeNoiseContextEncoder