Unfortunately, they didn't give many details in the paper. It's frustrating, to say the least. Yay reproducibility. They say > We use the same model and archit

Unfortunately, they didn't give many details in the paper. It's frustrating, to say the least. Yay reproducibility. They say > We use the same model and architecture as GPT-2 [RWC+19], including the modified initialization, pre-normalization, and reversible tokenization described therein, with the exception that we use alternating dense and locally banded sparse attention patterns in the layers of the transformer, similar to the Sparse Transformer. In the referenced paper (sparse transformer) they showed a bunch of different sparsity patterns, and I believe they're referring to either their banded block diagonal sparsity or a true banded diagonal pattern (local windows like bigbird and some other papers). Unfortunately, that paper also was light on details and the repo they open sourced alongside it is inscrutible.