layernorm.mojo Reference: Layer Normalization Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift layernorm.c