Weight initialization schemes for PyTorch nn.Modules - View it on GitHub