call_init(fn: 'InitFn', t: 'Tensor', **kwargs: 'DepthIndex') -> 'None'
kaiming_uniform(w: 'Tensor', *, depth_index: 'DepthIndex' = ()) -> 'None'
kaiming_normal(w: 'Tensor', *, depth_index: 'DepthIndex' = ()) -> 'None'
xavier_uniform(w: 'Tensor', *, depth_index: 'DepthIndex' = ()) -> 'None'
xavier_normal(w: 'Tensor', *, depth_index: 'DepthIndex' = ()) -> 'None'
normal(w: 'Tensor', *, std: 'float' = 0.02, depth_index: 'DepthIndex' = ()) -> 'None'
truncated_normal(w: 'Tensor', *, std: 'float' = 0.02, depth_index: 'DepthIndex' = (), lower: 'float' = -2.0, upper: 'float' = 2.0, variance_correction: 'bool' = False) -> 'None'
unit_fan_in_uniform(w: 'Tensor', *, depth_index: 'DepthIndex' = ()) -> 'None'
mup_output(w: 'Tensor', *, depth_index: 'DepthIndex' = ()) -> 'None'
dirac(w: 'Tensor') -> 'None'
