Normalization
=============

.. currentmodule:: attnax

.. autosummary::

   RMSNorm
   create_norm

.. autoclass:: RMSNorm

.. autofunction:: create_norm