[hk] Allow variance scaling initializer of 0.
[hk] Allow variance scaling initializer of 0.

PiperOrigin-RevId: 342648172
Change-Id: I5701af589ced11fb86a0739047bee17e611e9900

deepmind/dm-haiku,deepmind/dm-haiku