flax

0347dd2f - Update `kernel_size` docstring for `nn.Conv{local}`.

Update `kernel_size` docstring for `nn.Conv{local}`. Passing an int for kernel_size has been forbidden since [PR 1451](https://github.com/google/flax/pull/1451). PiperOrigin-RevId: 570940226