Simple implementation of the LSUV initialization in PyTorch - View it on GitHub