Searching protocol for "LayerNorm"
Stabilize deep networks, accelerate training.
Train modular MLP backbones with Flax NNX.