LLM Training: The End-to-End Flow
What each step of an LLM training iteration computes, what it costs in memory and arithmetic, and where the loop stalls on real hardware.
What each step of an LLM training iteration computes, what it costs in memory and arithmetic, and where the loop stalls on real hardware.
LLM Basic
What the KV cache costs per token, and how MQA, GQA and MLA rewrite attention to make that number smaller.
The two data-movement primitives behind training performance: tiling inside a device, collectives between devices.
The mechanism behind step 5 of the training loop: bucketed gradient all-reduce triggered from autograd hooks, what NCCL does with it, and how to diagnose the result.
How PagedAttention uses virtual-memory-style KV-cache paging to reduce fragmentation and increase LLM serving throughput.
What each of the standard building blocks actually does, why it replaced the thing before it, and what it costs.
How FlashAttention computes exact attention without materializing quadratic score and probability matrices in GPU memory.
What quantization actually changes in a network, which parts of a transformer break first, and which techniques repair them.
How successive FlashAttention implementations respond to the bottleneck exposed by each new GPU generation.
FlashInfer treats LLM attention as a serving systems problem spanning data layout, kernel generation, and runtime scheduling.
How agentic orchestration and Mixture-of-Experts routing introduce conditional execution at different levels of the LLM inference stack.