cs336-coach

cs336-coach is a skill for Cursor from JingyaLiu/ml-rs-interview-agent. It costs 69 tokens per session (691 once invoked), scanned A, original, MIT.

A study coach for Stanford's CS336 course on building language models, including tokenization, transformers, attention, and training code. It also supports the course's first programming assignment and related practice files.

In plain words
What is it for?
It is for reviewing lectures, practising machine-learning code from scratch, preparing for interviews, and working through CS336 Assignment 1.
Why use it?
It turns lecture material and coding exercises into structured practice while keeping core implementation work as guided learning rather than completed answers.

Skill for Cursor

Written for Cursor: installed under .cursor/.

Good fit It is for reviewing lectures, practising machine-learning code from scratch, preparing for interviews, and working through CS336 Assignment 1.

Compare 6 skills from other repositories ↓
Install with agentmods
npx agentmods add skills/jingyaliu/ml-rs-interview-agent/cs336-coach
Install

Getting it into your agent

One page per mod, every tool's command on it. A separate URL per tool would split the same page into five that compete with each other.

Any agent
npx skills add JingyaLiu/ml-rs-interview-agent --skill cs336-coach
Clone the repo
git clone --depth 1 https://github.com/JingyaLiu/ml-rs-interview-agent

Made for: Cursor.

Wrote this? Show the measurements

A badge with what this costs and how it scanned, read live from this page, so it follows the numbers instead of freezing them. Markdown for a README, HTML for a documentation site or a project page.

agentmods badge for cs336-coach

README.md
[![agentmods](https://agentmods.dev/badge/skills/jingyaliu/ml-rs-interview-agent/cs336-coach/github.svg)](https://agentmods.dev/skills/jingyaliu/ml-rs-interview-agent/cs336-coach)
Your own site
<a href="https://agentmods.dev/skills/jingyaliu/ml-rs-interview-agent/cs336-coach"><img src="https://agentmods.dev/badge/skills/jingyaliu/ml-rs-interview-agent/cs336-coach/github.svg" alt="Measured on agentmods" height="20"></a>

Or the 80×15 button, for a site that already has a row of RSS and ATOM ones. Only the verdict fits; the numbers stay here.

agentmods 80×15 button for cs336-coach

Your own site · 80×15
<a href="https://agentmods.dev/skills/jingyaliu/ml-rs-interview-agent/cs336-coach"><img src="https://agentmods.dev/badge/skills/jingyaliu/ml-rs-interview-agent/cs336-coach.svg" alt="Reviewed on agentmods" width="80" height="20"></a>
Per session 69 Skills are progressive disclosure: only the name and description are preloaded; the body loads when the skill is used.
When invoked 691 The whole file, excluding the scripts and references it only reads on demand.
Security scan A 0 findings. A grade says what 26 rules found in the file — not that it is safe.
Origin original No closer match found in the catalogue.
Token cost

What it costs to keep this loaded

Counted locally with the o200k_base tokenizer, which is exact for GPT models; Claude uses its own tokenizer and its counts differ. Treat this as one consistent yardstick across the catalogue rather than a bill. Prices are per million input tokens.

ModelPer sessionOnce invoked
Fable 5.1 $0.00069 $0.00691
Opus 5 $0.00034 $0.00345
Sonnet 5 $0.00014 $0.00138
Haiku 4.5 $0.00007 $0.00069

Measured 12d ago against content hash 2a2e4d06f265, method: parsed. Prices are Anthropic first-party input rates as of 2026-09-12, from the pricing page.

Security

Grade A, and why

cs336-coach scanned grade A with 0 findings against 26 rules in 11 categories — prompt injection, anti-refusal, data exfiltration, privilege escalation, supply chain, agent snooping, system-prompt leakage, SSRF and excessive agency — measured 12d ago.

A static scan of the body, not an audit. Every finding is printed with the line that produced it so you can judge whether it matters here. A mod is markdown that instructs an agent; that is exactly why what it instructs is worth reading.

Nothing flagged

None of the 26 patterns this scan looks for appear in this file: no shell pipes, no recursive deletes, no credential paths, no hidden text, no instruction-override or anti-refusal phrasing, no agent-config snooping. That is not a guarantee, it is the absence of the things that are checkable.

.cursor/skills/cs336-coach/SKILL.md · 78 lines

How it starts

The opening of the file, as written. The whole thing — 78 lines — stays where its author put it; the contents beside it link to each section on GitHub.

CS336 / LLM Coach

Goal

Turn lectures + assignment into interview-ready intuition and from-scratch coding speed (attention/MHA ≤45 min).

Respect Stanford CS336 agent guidelines when working in assignment repos: hints and review, not full solutions for core implementations.

Paths

Resource Path
Lecture notes Learning-Vault/ml-coding/cs336-notes.md
Drills Learning-Vault/ml-coding/drills/
Practice log Learning-Vault/ml-coding/practice-log.md
Lecture scripts Learning-Vault/ml-coding/cs336-materials/ (optional)
Assignment 1 cs336-assignment1-basics/ (sibling or linked repo)
Lectures repo cs336-lectures/ (optional)
Tests / adapters cs336-assignment1-basics/tests/

Modes (pick one per turn)

A. Lecture mode

  1. Confirm lecture # / topic
  2. Deliver or elicit 3 takeaways + 1 diagram-from-memory ask
  3. Append concise notes to cs336-notes.md
  4. Link to matching drill or assignment section

B. Drill mode (whiteboard ML coding)

  • Files: ml-coding/drills/*.py
  • Hints only until a genuine attempt (same ladder as leetcode-coach)
  • Timed goal: attention / MHA from blank ≤45 min
  • Log result in practice-log.md

C. Assignment mode

  1. Read failing test + tests/adapters.py
  2. Guide implementation in cs336_basics/ — do not paste complete solutions for core components
  3. Explain why: shapes, complexity, numerical stability
  4. Verify with uv run pytest when available

Interview mapping

CS336 topic Interview use
BPE / tokenizer systems + coding
Attention / MHA whiteboard ML coding
Training / optim research discussion
Parallelism (later) systems depth (stretch roles)

Shape checklist (always for tensor code)

  • Batch · seq · heads · d_model · d_head — say them aloud
  • Causal mask direction
  • Softmax axis
  • Train vs inference differences when relevant

Example prompts

Say this Expect
Explain multi-head attention shapes like an interview Shape-first walkthrough
I'm stuck on Assignment 1 BPE — guide don't code Test-driven hints
Quiz me on Lec 3 decoder stack from memory Socratic quiz → notes gap

Read the full file on GitHub · 78 lines

Changes

What this file has done since we first saw it

Hashed on every crawl. A supply-chain change to an agent config is a question of when, not whether, so the history is kept rather than the latest state alone.

  1. 12d ago First seen · 78 lines · 69 tokens per session scan A 2a2e4d06f265

Subscribe to this mod's changes

cs336-coach is a skill published in the GitHub repository JingyaLiu/ml-rs-interview-agent (5 stars, last pushed 1mo ago), licensed MIT. It adds 69 tokens to every session and 691 once invoked, about $0.0003 per session on Opus 5. A static security scan graded it A with 0 findings. No closer match exists in the catalogue, so it is treated as the original; first seen 2026-08-31.

Related

Other skills, from other repositories

openai-docs

Use when the user asks how to build with OpenAI products or APIs and needs up-to-date official documentation with citations, help choosing the latest model for a use case, or model upgrade and prompt-upgrade guidance; prioritize OpenAI docs MCP tools, use bundled references only as helper context, and restrict any…

Haohao-end/openagent · 74 tokens

hr-prompt-engineering

Help HR professionals, recruiters, and people teams write better AI prompts to get higher-quality, more accurate, and more useful outputs from AI tools across HR workflows. Use when asked to write better AI prompts, improve my prompt for HR tasks, learn prompt engineering for HR, get better ChatGPT outputs, design…

tuanductran/hr-skills · 83 tokens

transformer-architecture-guide

Guide to Transformer architectures for NLP and computer vision.

wentorai/research-plugins · 15 tokens

minimind-learning

A Chinese-language learning assistant for MiniMind, a small language-model project, that records study notes and recognizes common machine-learning terms.

joyehuang/minimind-notes · 76 tokens

prompt-engineering-interviewer

A Senior AI Engineer interviewer that simulates a technical interview focused on prompt engineering and LLM architecture at scale. Use this agent when you want to practice prompt pipeline design, RAG architecture, evaluation frameworks, token optimization, and edge case handling. This evaluates engineering rigor and…

PrepLabsAI/InterviewMentor · 70 tokens

nexus-tutorial

Use for creating executable Jupyter tutorials and AI engineering walkthroughs with runnable cells. Trigger on requests for step-by-step guides, notebook-based teaching, or shareable code-first learning content. Prioritize reproducibility, clarity, and copy-paste-ready outputs. When in doubt, use this skill.

aayushostwal/nexus · 64 tokens