CS336 Assignment 1: Large Language Model Training and Inference

Stanford CS336 Assignment 1 is titled Building a Transformer LM . It covers the main ideas behind a decoder-only language model: