A LARS implementation in PyTorch - View it on GitHub