Paper

And Scaling Transformer to 1M tokens and beyond with RMT