dmx.compressor.modeling.nn.custom\_modules
==========================================

.. automodule:: dmx.compressor.modeling.nn.custom_modules
   :members:
   :undoc-members:
   :show-inheritance:

   
   .. rubric:: Functions

   .. autosummary::
      :toctree:
   
      apply_rotary_embeddings
   
   .. rubric:: Classes

   .. autosummary::
      :toctree:
   
      ApplyRotaryPosEmb
      ApplyRotaryPosEmbBase
      BloomGELU
      ClippedGELU
      FastGELU
      GemmaRMSNorm
      NewGELU
      QuickGELU
      RotaryEmbedding