Fix the order of layernorm and residual for postnorm. Check https://arxiv.org/pdf/2002.04745.pdf Figure 1 for details.
Fix the order of layernorm and residual for postnorm.
Check https://arxiv.org/pdf/2002.04745.pdf Figure 1 for details.

PiperOrigin-RevId: 457586835

google/praxis,google/praxis