Build Your Own ML Framework
🚧 Preview · Classroom ready 2026
Hands-on labs for the Machine Learning Systems textbook
Don't import it. Build it.
From tensors to systems. An educational framework for building and optimizing ML—understand how PyTorch, TensorFlow, and JAX really work.
TinyTorch: Build AI Like Bricks
Recreate ML History#
Walk through ML history by rebuilding its greatest breakthroughs with YOUR TinyTorch implementations. Click each milestone to see what you’ll build and how it shaped modern AI.
Why Build Instead of Use?#
Traditional ML Education
import torch
model = torch.nn.Linear(784, 10)
output = model(input)
# When this breaks, you're stuck
Problem: You can't debug what you don't understand.
TinyTorch: Build → Use → Reflect
# BUILD it yourself
class Linear:
def forward(self, x):
return x @ self.weight + self.bias
# USE it on real data
loss.backward() # YOUR autograd
Advantage: You can debug it because you built it.
Learning Path#
Four progressive tiers take you from foundations to production systems:
Foundation (01-07)
Tensors, autograd, layers, training loops
Architecture (08-13)
CNNs, attention, transformers, GPT
Optimization (14-19)
Profiling, quantization, acceleration
Torch Olympics (20)
Competition-ready capstone project
The Big Picture • Getting Started • Preface
Is This For You?#
🎓 Students
Taking ML courses, want to understand what's behind import torch
👩‍🏫 Instructors
Teaching ML systems with ready-made hands-on labs
🚀 Self-learners
Career changers or hobbyists going deeper than tutorials
Prerequisites: Python + basic linear algebra. No ML experience required.
Join the Community#
See learners building ML systems worldwide
Add yourself to the map • Share your progress • Connect with builders
Next Steps: Quick Start (15 min) • The Big Picture • Community