Skip to main content

burn_mamba/modules/norm/
mod.rs

1/// Root-mean-square normalisation (last-dim, fp16-safe); also the Mamba-3
2/// QK-Norm.
3pub mod rms_norm;
4/// RMSNorm followed by a SiLU(z) gate (Mamba-2 output norm).
5pub mod rms_norm_gated;