zea.models.dit
==============

.. automodule:: zea.models.dit

   
   .. rubric:: Functions

   .. autosummary::
   
      get_time_conditional_dit_network
      modulate
   
   .. rubric:: Classes

   .. autosummary::
   
      AddPositionEmbedding
      DiTBlock
      DiTTimeConditional
      FinalLayer
      Unpatchify