TorchLeet

LeetCode for PyTorch — 65 ML/AI interview problems from real interviews at Google, Meta, Anthropic. Jupyter notebooks, an auto-grader, and an MCP AI tutor.

What is TorchLeet?

LeetCode for PyTorch — 65 ML/AI interview problems from real interviews at Google, Meta, Anthropic. Jupyter notebooks, an auto-grader, and an MCP AI tutor.

Overview of TorchLeet

What it is

TorchLeet is a collection of 68 PyTorch problems sourced from real ML/AI interviews at companies including Google, Meta, Anthropic, OpenAI, DeepMind, and others. It provides Jupyter notebooks for each problem, an auto-grader to check solutions, and an MCP (Model Context Protocol) AI tutor that can guide learners through problems without giving away answers. The problems are organized into three tracks: Basics (24 problems), LLM Learning Path (23 problems), and Advanced (48 problems), with company tags indicating which problems are asked by specific employers.

Key features

  • 68 real interview problems from companies like Google, Meta, Anthropic, OpenAI, DeepMind
  • Three tracks: Basics, LLM Learning Path (build an LLM from scratch), and Advanced
  • Jupyter notebooks with question files and solution files (marked _SOLN)
  • AI Tutor via MCP server that provides progressive hints, company prep plans, and learning guides (torchleet-tutor, torchleet-interview-prep, torchleet-review, torchleet-explain)
  • Company-tagged questions so users know exactly what each company asks
  • Auto-grader to verify solutions

Use cases

  • Preparing for ML/AI engineering interviews at top tech companies
  • Building a complete LLM from scratch through a structured learning path
  • Practicing core PyTorch fundamentals and advanced topics like attention mechanisms, alignment techniques, and GPU kernels

Getting started

bash# Clone the repogit clone https://github.com/Exorust/TorchLeet.gitcd TorchLeet# Install PyTorch (see https://pytorch.org/get-started/locally/)# Open a problem notebookjupyter notebook torch/basic/lin-regression/lin-regression.ipynbTo use the AI Tutor, connect via MCP:bash# Claude Codeclaude mcp add torchleet -- npx -y torchleet-mcp# Codexcodex mcp add torchleet -- npx -y torchleet-mcpFor Claude Desktop / Cursor / VS Code, add the MCP server configuration as described in the README.

License and pricing

This project is licensed under the MIT License and is open source (free).

Based on the project's public README and repository metadata.

Add this badge to your README

Featured on YunoDB Badge styles & help

Maintain this project? Show that it is featured on YunoDB.

Markdown[![Featured on YunoDB](https://yunodb.com/badge/exorusttorchleet.svg)](https://yunodb.com/tools/exorusttorchleet?utm_source=github&utm_medium=badge)
HTML<a href="https://yunodb.com/tools/exorusttorchleet?utm_source=github&utm_medium=badge"><img src="https://yunodb.com/badge/exorusttorchleet.svg" alt="Featured on YunoDB" height="20"></a>