If it’s not too difficult, it would be nice to have in nn.init. It’ll make it much easier for people to drop in and see if it helps for their problems, and we can also link PixelShuffle to it so that people are aware that there’s a principled weight initialisation available.