KaimingUniform

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