VarianceScaling

class brainstate.nn.init.VarianceScaling(scale, mode, distribution, in_axis=-2, out_axis=-1, seed=None, unit=Unit('1'))