pymllm.layers.layer_norm
========================

.. py:module:: pymllm.layers.layer_norm


Classes
-------

.. autoapisummary::

   pymllm.layers.layer_norm.LayerNorm


Module Contents
---------------

.. py:class:: LayerNorm(hidden_size, eps = 1e-06)

   Bases: :py:obj:`pymllm.layers.base.MllmBaseLayer`


   LayerNorm layer implemented with FlashInfer kernel.


   .. py:attribute:: hidden_size


   .. py:attribute:: eps
      :value: 1e-06



   .. py:attribute:: weight


   .. py:attribute:: bias


   .. py:method:: forward(x)