TinyTorch: Building Machine Learning Systems from First Principles
Not on the map. Why?
Abstract (arXiv)
Machine learning education faces a fundamental gap: students learn algorithms without understanding the systems that execute them. They study gradient descent without measuring memory, attention mechanisms without analyzing O(N^2) scaling, optimizer theory without knowing why Adam requires 3x the memory of SGD. This "algorithm-systems divide" produces practitioners who can train models but cannot debug memory failures, optimize inference latency, or reason about deployment trade-offs--the very skills industry demands as "ML systems engineering." We present TinyTorch, a 20-module curriculum that closes this gap through "implementation-based systems pedagogy": students construct PyTorch's core components (tensors, autograd, optimizers, CNNs, transformers) in pure Python, building a complete framework where every operation they invoke is code they wrote. The design employs three patterns: "progressive disclosure" of complexity, "systems-first integration" of profiling from the first module, and "build-to-validate milestones" recreating 67 years of ML breakthroughs--from Perceptron (1958) through Transformers (2017) to MLPerf-style benchmarking. Requiring only 4GB RAM and no GPU, TinyTorch demonstrates that deep ML systems understanding is achievable without specialized hardware. The curriculum is available open-source at mlsysbook.ai/tinytorch.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 10 of its 46 references that are in the map. First posted on arXiv 27 Jan 2026.
Builds on
10 of 46 references.
References · 10
- 2017Attention Is All You Needcited by 60,361
- 2019Energy and Policy Considerations for Deep Learning in NLPcited by 1,254
- 2021Carbon Emissions and Large Neural Network Trainingcited by 567
- 2019MLPerf Inference Benchmarkcited by 199
- 2020fastai: A Layered API for Deep Learningcited by 183
- 2019MLPerf Training Benchmarkcited by 115
- 2021Dive into Deep Learningcited by 111
- 2020Benchmarking TinyML Systems: Challenges and Directioncited by 91
- 2023ASTRA-sim2.0: Modeling Hierarchical Networks and Disaggregated Systems for Large-model Training at Scalecited by 47
- 2021Widening Access to Applied Machine Learning with TinyMLcited by 17
Built on by
Citing papers per year: 2026: 1.
1 paper.