Implementation of N-Grammer, augmenting Transformers with latent n-grams, in Pytorch - View it on GitHub