This repo includes a Sparse Transformer implementation which utilizes PQ to derive the sparsity.