Python module
max.experimental.nn.norm
Normalization layers for MAX neural networks.
Normalization layers
GemmaRMSNorm | Computes the Root Mean Square normalization on inputs. |
|---|---|
GroupNorm | Group normalization block. |
LayerNorm | Layer normalization over the last dimension. |
RMSNorm | Computes the Root Mean Square normalization on inputs. |
Functions
rms_norm | Applies Root Mean Square layer normalization to an input tensor. |
|---|
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!