Official codebase for Pretrained Transformers as Universal Computation Engines. - View it on GitHub