mstar.model.waypoint.components.layers
======================================

.. automodule:: mstar.model.waypoint.components.layers

   
   
   

   
   
   .. rubric:: Functions

   .. autosummary::
   
      ada_gate
      ada_rmsnorm
      bf16_roundtrip
      fp32_matmul_highest
      rms_norm
   
   

   
   
   .. rubric:: Classes

   .. autosummary::
   
      AdaLN
      CondHead
      ControllerInputEmbedding
      DeviceTableCache
      MLP
      MLPFusion
      NoiseConditioner