Keyboard shortcuts

Press or to navigate between chapters

Press S or / to search in the book

Press ? to show this help

Press Esc to hide this help

Phase 14 — Exercises: Model Architectures (Adding a Model)

Work these after the labs. They escalate from "explain it" to "design it" — staff-level means you can do the last ones cold.

  1. Map a HF attention block's qkv/o weights onto QKVParallelLinear/RowParallelLinear.
  2. What must change to make a model support tensor parallelism correctly?
  3. How would you add a pooling/reward head, and what changes in output handling?

Self-grading

For each: could you (a) explain it to a teammate in 2 minutes, and (b) point to the exact upstream/ file that proves your answer? If not, re-read the matching anchor in 01-deep-dive.md.