flax

a7b0edeb - Don't create param in normalization layers instead of create None-value params.

Don't create param in normalization layers instead of create None-value params. This makes these layers align better with behavior of Linen layers, and also reduce confusion. PiperOrigin-RevId: 719498793