XavierUniform

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