Imported from https://github.com/fab-jul/L3C-PyTorch