- Front
- Dedication
- Foreward
- Preface
- I. Structure and Intepretation of Tensor Programs
- 1. Elements of Learning
- 1.1. Types to Tensors
- 1.2. Probabilities as Tensors
- 1.3. Linear Models and their Algebra
- 1.4. Non-Linear Models and their Stochastic Optimizers
- 2. Anatomy of an Autograd
2.1. Forwards
2.2. Backwards
2.3. Double Backwards
2.4. Mutation
2.5. Streams
- II. Compilation of Tensor Programs
- 3. Serial Compilation
- 3.1. Parsing llm.c (gpt2) to AST
- 3.2. CFG-SSA + Linear Scan
- 3.3. SoN + Graph Coloring
- 4. Parallel Compilation
- 5. Differentiable Compilation
- Afterword
- Bibliography
- Index