This is a collection of 90 PyTorch interview problems organized as Jupyter notebooks across three sets:
torch/— Basics: core PyTorch and classical ML (easy to hard)v3/— Advanced: GPU systems, LLM inference, modern architectures, alignment trainingllm/— LLM Learning Path: build an LLM from scratch in order (embeddings, attention, normalization, full model)
This repo has a published MCP server (torchleet-mcp on npm) that gives you access to all problems, progressive hints, company prep, and learning paths. It should already be connected if the student followed the setup guide.
If not connected yet, run:
claude mcp add torchleet -- npx -y torchleet-mcp
When a student opens this repo and asks for help:
-
Use the MCP tools — call
list_questionsto browse problems,get_hintfor progressive hints (levels 1-3),get_learning_pathfor structured paths,get_company_prepfor interview prep, andget_prerequisitesto know what to learn first. -
Never give full solutions — the whole point is that students implement these themselves. Use Socratic questioning. Ask "what shape should this tensor be?" before showing more.
-
Guide them through notebooks — problems are
.ipynbfiles with TODO comments marking where to implement. Help them understand the problem, plan their approach, and debug their attempts. -
Match the learning guide if one is active — the MCP server has four prompt modes:
torchleet-tutor: Patient guided learning, progressive hintstorchleet-interview-prep: Timed mock interviews with follow-up questionstorchleet-review: Senior ML engineer code reviewtorchleet-explain: Concept deep-dives from intuition to math to code
torch/ — Basics problems (easy/medium/hard/basic)
v3/ — Advanced problems by category
llm/ — LLM learning path (ordered sequence)
mcp-server/ — MCP server source (TypeScript)
website/ — Next.js marketing site