# Cs336 Coach

> Guides Stanford CS336 LLM systems learning, from-scratch drills, and Assignment 1 implementation. Use when studying lectures, tokenization/BPE, transformers, attention/MHA, training loops, einops, or working in cs336-assignment1-basics / ml-coding drills / cs336-notes.

- **Type:** Skill
- **Install:** `agentstack add skill-jingyaliu-ml-rs-interview-agent-cs336-coach`
- **Verified:** Yes — security-reviewed for prompt injection and unsafe behavior
- **Seller:** [JingyaLiu](https://agentstack.voostack.com/s/jingyaliu)
- **Installs:** 0
- **Category:** [AI & ML](https://agentstack.voostack.com/c/ai-and-ml)
- **Latest version:** 0.1.0
- **License:** MIT
- **Upstream author:** [JingyaLiu](https://github.com/JingyaLiu)
- **Source:** https://github.com/JingyaLiu/ml-rs-interview-agent/tree/main/.cursor/skills/cs336-coach

## Install

```sh
agentstack add skill-jingyaliu-ml-rs-interview-agent-cs336-coach
```

Requires the [AgentStack CLI](https://agentstack.voostack.com/docs/cli). Works with Claude Code, Cursor, and any MCP-compatible agent.

## About

# 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 |

## Style

- Socratic for concepts; concrete for shapes/dtypes
- One concept per teaching turn
- Tie blocks to user's focus pillars in `profile.md` when useful

## Source & license

This open-source skill is cataloged on AgentStack and links to its original source — we do not rehost the code.

- **Author:** [JingyaLiu](https://github.com/JingyaLiu)
- **Source:** [JingyaLiu/ml-rs-interview-agent](https://github.com/JingyaLiu/ml-rs-interview-agent)
- **License:** MIT

Install and usage instructions live in the source repository linked above.

## Pricing

- **Free** — Free

## Security capabilities

Automated source analysis of v0.1.0 — what this tool can access:

- **Network access:** no
- **Filesystem access:** no
- **Shell / process execution:** no
- **Environment & secrets:** no
- **Dynamic code execution:** no

*"Yes" means the capability is present in the source — more access means more to trust, not that it is unsafe.*


## Versions

- **0.1.0** — security scan: passed — Imported from the upstream source.

## Links

- Listing page: https://agentstack.voostack.com/l/skill-jingyaliu-ml-rs-interview-agent-cs336-coach
- Seller: https://agentstack.voostack.com/s/jingyaliu
- Browse the marketplace: https://agentstack.voostack.com/browse

---
Listed on AgentStack — the marketplace for AI agent skills and MCP servers. Every listing is security-reviewed. Creators keep 70%.