Pytorch implementation of Compressive Transformers, from Deepmind - View it on GitHub