LogoCookLLM Docs
LogoCookLLM Docs
HomeCookLLM

Principles

Tokenization
Tokenization BasicsBPE AlgorithmGPT TokenizersBPE Training Engineering
Model Architecture
Transformer LM
From token ids to logitsEmbedding and LM Head
Attention Mechanisms
From Self-Attention to GQAAttention Sink
Position Encoding
Position Encoding BasicsRoPE Math DerivationRoPE ImplementationLength Extrapolation
GPU Programming Basics
GPU Architecture BasicsTensor LayoutTriton Basics: Vector Add
FlashAttention
Flash Attention PrinciplesFrom Naive to Auto-TuningBlock Pointers and Multi-Dim SupportCausal Masking OptimizationGrouped Query AttentionBackward Pass
Distributed Training
Data ParallelismZeRO OptimizerFully Sharded Data Parallel张量并行流水线并行多维混合并行

Hands-on Training

Overview
Pretraining
Pretraining DataTokenizer TrainingModel ArchitectureData PipelineTraining LoopMonitoring and Validation
X (Twitter)
SystemsDistributed Training

Data Parallelism

Premium

Understanding communication primitives and DDP's gradient synchronization mechanism

Companion Code

Log in to continue reading

This is premium content. Please log in to access the full article.

Distributed Training

From data parallelism to multi-dimensional hybrid parallelism — understanding the core parallel strategies of large model training

ZeRO Optimizer

Progressive de-redundancy: three-stage sharding from optimizer states to parameters

Table of Contents

Memory Composition of Single-GPU Training
Memory Requirements of Mixed-Precision Training
Communication Primitives
Broadcast
All-Reduce
Reduce-Scatter
All-Gather
DataParallel: The Most Naive Multi-GPU Approach
How DDP Works
Ring All-Reduce: Efficient Gradient Synchronization
Gradient Synchronization Mechanism
Limitations of DDP
Summary