flax

Removed kernel_init/bias_init atttributes from popular layers

#4998

Merged