Speed up StableNorm for non-trivial sizes and improve consistency between aligned and unaligned inputs.