def METHOD_NAME(module):
    def normal_(data):
        data.copy_(data.normal_(mean=0.0, std=0.02).to(data.device))
    if isinstance(module, nn.Linear):
        normal_(module.weight.data)
        if module.bias is not None:
            module.bias.data.zero_()
    elif isinstance(module, nn.Embedding):
        normal_(module.weight.data)
        if module.padding_idx is not None:
            module.weight.data[module.padding_idx].zero_()
    elif isinstance(module, MultiwayNetwork):
        normal_(module.A.weight.data)
        normal_(module.B.weight.data)