cs.AIApr 28, 2026

Training Transformers as a Universal Computer

Authors: Ruize Xu, Chenxiao Yang, Yanhong Li, David McAllester

Organizations: The University of Chicago · Toyota Technological Institute at Chicago · Allen Institute for AI

Abstract

We demonstrate that a small transformer can learn to execute programs in MicroPy, a simplified yet computationally universal programming language. Given procedure definitions together with an expression to evaluate, the transformer predicts small-step execution using PENCIL scaffolding for space-efficient execution within a bounded context window. After training on randomly generated, meaningless MicroPy programs, the learned transformer generalizes to various human-written programs including bit copying and flipping, binary addition and multiplication, and SAT verification and solving. We note that the trained model can achieve out-of-distribution generalization; i.e., evaluate novel programs from distribution on programs. Since MicroPy can express any computation, our results provide empirical evidence that a standard transformer can be trained to act as a universal computer.

Explore similar work

CardsList
  1. Fixed Universal Transformers

    May 29, 2026Jingwen Liu, Alexandr Andoni, Daniel HsuTransformer ArchitecturesUnified Transformer