Concept atlas

Foundations of Modern Machine Learning

100 concepts behind modern machine learning, from maximum likelihood and attention to generative models, interpretability and alignment. Each page gives a short summary, the key equation and the math behind it, key papers and links to related concepts.

100 concepts40 with demos330 links13 study phases
1Core training
2Optimization
3Generative models
4Represen­tations
5Scaling and alignment
6Systems
interactive demospredict, drag, check

Study guide

Start with the map when you are unsure what to learn first.

This atlas100 concepts and the links between them205 prerequisite links · 125 other relations · 40 with interactive demos · 13 study phases
Question

Which simpler idea do I need first to understand the method I am stuck on?

How to use this mapPredict · Check · Generalize · Continue
PredictGuess where the gap is.

Before opening an advanced page, guess whether what blocks you is notation, optimization, probability, representations, scaling or systems.

CheckUse the map, the papers and the demos together.

The map shows which concepts build on which; each page cites its key papers; 40 pages include an interactive demo.

GeneralizeLook for ideas that recur.

The most useful concepts explain several models, papers or engineering trade-offs at once.

ContinueChoose the map, the study order or the list.

Browse the map for neighboring concepts, follow the study order when sequence matters, or filter the list below to pages with demos.

Recommended Study Order

A suggested reading order, from fundamentals to recent research. Most concepts come after all of their prerequisites; some depend on a concept listed later.

1

Core probabilistic training and transformers

11

Reasoning at inference time & synthetic data

All Concepts

Search by concept name, equation, paper title or author.

1

Maximum Likelihood

Maximum Likelihood, Cross-Entropy & KL Divergence

Core Training27 links2 sources▶ Demo
Open the reference →
2

Attention

Scaled Dot-Product Attention & Transformer Layers

Core Training23 links3 sources▶ Demo
Open the reference →
3

Adam

Adam & Adaptive Gradient Methods

Optimization4 links2 sources▶ Demo
Open the reference →
4

Sharpness

Loss Landscapes, Sharpness & Flat Minima

Optimization11 links4 sources▶ Demo
Open the reference →
5

Double Descent

Overparameterization & Generalization, Double Descent

Optimization4 links4 sources▶ Demo
Open the reference →
6

NTK

Neural Tangent Kernel & Infinite-Width Limits

Theory3 links3 sources▶ Demo
Open the reference →
7

VAEs

Variational Autoencoders & Variational Inference

Generative Models5 links2 sources▶ Demo
Open the reference →
8

GANs

GANs & Adversarial Divergence Minimization

Generative Models2 links2 sources▶ Demo
Open the reference →
9

Diffusion

Diffusion, Score-Based Models & Flow Matching

Generative Models10 links4 sources▶ Demo
Open the reference →
10

Embeddings

Representation Learning & Embedding Geometry

Representations14 links3 sources▶ Demo
Open the reference →
11

Superposition

Superposition, Sparse Features & Monosemanticity

Representations3 links2 sources▶ Demo
Open the reference →
12

Probing

Probing, Linear Classifier Probes & Activation Analysis

Representations5 links3 sources▶ Demo
Open the reference →
13

Circuits

Transformer Circuits, Induction Heads & Mechanistic Interpretability

Representations7 links3 sources▶ Demo
Open the reference →
14

Scaling Laws

Scaling Laws & Emergent Abilities

Scaling & Alignment7 links4 sources▶ Demo
Open the reference →
15

RLHF

Preference-Based Alignment: RLHF, Reward Modeling, Constitutional AI

Scaling & Alignment17 links4 sources▶ Demo
Open the reference →
16

Efficiency

Efficiency: Quantization, Distillation, LoRA & Sparse MoE

Efficiency8 links4 sources▶ Demo
Open the reference →
17

Learning Theory

Theoretical Foundations: PAC Learning, MDL & Information Bottleneck

Theory9 links7 sources▶ Demo
Open the reference →
18

RoPE

Rotary Position Embeddings (RoPE)

Representations3 links4 sources▶ Demo
Open the reference →
19

Efficient Attention

Efficient Attention at Scale: KV Cache, GQA & FlashAttention

Efficiency11 links4 sources▶ Demo
Open the reference →
20

Speculative Decoding

Speculative Decoding: Lossless Multi-Token Generation

Efficiency5 links4 sources▶ Demo
Open the reference →
21

LLM Serving

LLM Serving at Scale: Prefill, Decode & Continuous Batching

Efficiency7 links4 sources▶ Demo
Open the reference →
22

MoE

Sparse Mixture of Experts: Routing, Load Balancing & Expert Parallelism

Efficiency8 links5 sources▶ Demo
Open the reference →
23

MoE Serving

MoE Serving & Scheduling: Token Dispatch, All-to-All, Disaggregated Parallelism

Efficiency4 links3 sources▶ Demo
Open the reference →
24

DPO

Direct Preference Optimization: RL-Free Alignment from Human Preferences

Scaling & Alignment5 links3 sources▶ Demo
Open the reference →
25

KTO

KTO: Alignment from Binary Feedback via Human-Aware Losses

Scaling & Alignment3 links3 sources▶ Demo
Open the reference →
26

Reward Hacking

Reward Hacking & Overoptimization: Goodhart's Law in Preference Optimization

Scaling & Alignment3 links5 sources▶ Demo
Open the reference →
27

Sparse Autoencoders

Sparse Autoencoders at Scale: Feature Dictionaries for Mechanistic Interpretability

Representations6 links4 sources▶ Demo
Open the reference →
28

Circuit Discovery

Automated Circuit Discovery: Patching, Attribution & Decomposition at Scale

Representations4 links4 sources▶ Demo
Open the reference →
29

Activation Steering

Activation Steering: Feature-Guided Interventions for Inference-Time Control

Representations4 links4 sources▶ Demo
Open the reference →
30

Long Context

Long Context Engineering: RoPE Scaling, KV Compression & Memory Optimization

Efficiency9 links4 sources▶ Demo
Open the reference →
31

State Space Models

State Space Models & Hybrid Architectures: Mamba-2, Jamba, Griffin

Core Training5 links8 sources▶ Demo
Open the reference →
32

Multimodal Pretraining

Multimodal Foundations: Vision Encoders, Contrastive Learning & Cross-Attention Fusion

Representations5 links6 sources▶ Demo
Open the reference →
33

Tokenization

Tokenization & Vocabulary Design

Representations2 links4 sources▶ Demo
Open the reference →
34

Decoding & Sampling

Decoding & Sampling: Temperature, Top-p & Inference-Time Control

Core Training3 links4 sources▶ Demo
Open the reference →
35

Backpropagation

Backpropagation & Automatic Differentiation

Optimization4 links3 sources
Open the reference →
36

Score Matching

Score Matching & Score-Based Generative Models

Generative Models5 links4 sources
Open the reference →
37

In-Context Learning

In-Context Learning: Learning Without Weight Updates

Representations5 links4 sources
Open the reference →
38

Optimal Transport

Optimal Transport & Wasserstein Distance

Theory3 links4 sources
Open the reference →
39

Normalizing Flows

Normalizing Flows: Exact Likelihood via Invertible Transforms

Generative Models3 links4 sources
Open the reference →
40

PPO

PPO: Proximal Policy Optimization

Scaling & Alignment1 link5 sources
Open the reference →
41

Residual Connections

Residual Connections & Skip Connections

Core Training2 links5 sources
Open the reference →
42

Classifier-Free Guidance

Classifier-Free Guidance in Diffusion

Generative Models2 links3 sources
Open the reference →
43

RAG

Retrieval-Augmented Generation (RAG)

Representations3 links4 sources
Open the reference →
44

Adversarial Examples

Adversarial Examples & Robustness

Theory3 links5 sources
Open the reference →
45

Grokking

Grokking: Delayed Generalization

Theory3 links3 sources
Open the reference →
46

Logit Lens

Logit Lens: Probing Intermediate Representations

Representations4 links2 sources
Open the reference →
47

Learning-Rate Schedules

Learning Rate Schedules: Warmup, Decay & Cycling

Optimization2 links4 sources
Open the reference →
48

Weight Initialization

Weight Initialization: Xavier, He & µP

Optimization3 links4 sources▶ Demo
Open the reference →
49

Contrastive Learning

Contrastive Learning & InfoNCE

Representations2 links4 sources
Open the reference →
50

Distributed Training

Distributed Training: Data, Tensor & Pipeline Parallelism

Efficiency3 links4 sources
Open the reference →
51

Beam Search

Beam Search & Structured Decoding

Core Training2 links5 sources
Open the reference →
52

Dropout

Dropout: Stochastic Regularization

Optimization2 links4 sources▶ Demo
Open the reference →
53

Energy-Based Models

Energy-Based Models & Score Functions

Generative Models2 links5 sources
Open the reference →
54

Layer Normalization

Layer Normalization & RMSNorm

Core Training2 links3 sources
Open the reference →
55

Fisher Information

Fisher Information & Information Geometry

Theory4 links4 sources
Open the reference →
56

Natural Gradient

Natural Gradient & Riemannian Optimization

Optimization3 links4 sources
Open the reference →
57

SGD & Momentum

Stochastic Gradient Descent & Momentum

Optimization2 links4 sources▶ Demo
Open the reference →
58

Weight Decay & AdamW

Weight Decay & AdamW: Decoupled Regularization

Optimization2 links3 sources
Open the reference →
59

Gradient Clipping

Gradient Clipping & Explosion Prevention

Optimization2 links3 sources
Open the reference →
60

Label Smoothing

Label Smoothing & Soft Targets

Optimization2 links4 sources
Open the reference →
61

Batch Normalization

Batch Normalization

Core Training1 link4 sources▶ Demo
Open the reference →
62

Knowledge Distillation

Knowledge Distillation: Learning from Teachers

Efficiency6 links4 sources
Open the reference →
63

Quantization

Quantization: Compressing Models to Integers

Efficiency2 links5 sources
Open the reference →
64

Pruning

Pruning: Removing Unnecessary Weights

Efficiency2 links6 sources
Open the reference →
65

Self-Supervised Learning

Self-Supervised Learning: Labels from Structure

Representations3 links3 sources
Open the reference →
66

Calibration

Calibration & Temperature Scaling

Theory3 links5 sources
Open the reference →
67

SwiGLU

SwiGLU & Gated Activations

Core Training1 link3 sources▶ Demo
Open the reference →
68

FlashAttention

FlashAttention: IO-Aware Attention

Efficiency2 links4 sources
Open the reference →
69

Constitutional AI

Constitutional AI: Principles-Based Alignment

Scaling & Alignment6 links4 sources
Open the reference →
70

Bregman Divergence

Bregman Divergence & Mirror Descent

Theory2 links4 sources
Open the reference →
71

Reproducing Kernels

Reproducing Kernel Hilbert Spaces

Theory2 links5 sources
Open the reference →
72

Persistent Homology

Persistent Homology & Topological Data Analysis

Theory2 links5 sources
Open the reference →
73

Lie Groups

Lie Groups & Equivariant Networks

Theory2 links6 sources
Open the reference →
74

Test-Time Compute

Test-Time Compute & Inference Scaling

Scaling & Alignment5 links4 sources
Open the reference →
75

Chain of Thought

Chain-of-Thought Prompting

Scaling & Alignment4 links6 sources
Open the reference →
76

World Models

World Models & Model-Based RL

Theory2 links4 sources
Open the reference →
77

Synthetic Data

Synthetic Training Data

Scaling & Alignment2 links5 sources
Open the reference →
78

Consistency Models

Consistency Models

Generative Models2 links3 sources
Open the reference →
79

Activation Checkpointing

Activation Checkpointing & Memory Efficiency

Efficiency2 links3 sources
Open the reference →
80

GQA

Grouped-Query Attention (GQA)

Efficiency2 links3 sources▶ Demo
Open the reference →
81

Process Reward Models

Process Reward Models

Scaling & Alignment2 links3 sources
Open the reference →
82

RLAIF

RLAIF: Reinforcement Learning from AI Feedback

Scaling & Alignment2 links2 sources
Open the reference →
83

Flow Matching

Flow Matching & Rectified Flows

Generative Models2 links3 sources
Open the reference →
84

Instruction Tuning

Instruction Tuning

Scaling & Alignment2 links3 sources
Open the reference →
85

Deliberative Alignment

Deliberative Alignment

Scaling & Alignment2 links2 sources
Open the reference →
86

Debate

AI Safety via Debate

Scaling & Alignment1 link2 sources
Open the reference →
87

Iterated Amplification

Iterated Amplification

Scaling & Alignment3 links2 sources
Open the reference →
88

Weak-to-Strong

Weak-to-Strong Generalization

Scaling & Alignment2 links1 source
Open the reference →
89

Automated Red Teaming

Automated Red Teaming

Scaling & Alignment2 links3 sources
Open the reference →
90

Mesa-Optimization

Mesa-Optimization & Inner Alignment

Scaling & Alignment2 links3 sources
Open the reference →
91

Sleeper Agents

Sleeper Agents & Alignment Faking

Scaling & Alignment2 links3 sources
Open the reference →
92

LLM-as-Judge

Model-Graded Evaluations

Scaling & Alignment2 links3 sources
Open the reference →
93

Capability Elicitation

Capability Elicitation & Eliciting Latent Knowledge

Scaling & Alignment2 links4 sources
Open the reference →
94

Sandwiching Evaluations

Sandwiching Evaluations

Scaling & Alignment2 links2 sources
Open the reference →
95

Mixture-of-Depths

Mixture-of-Depths

Efficiency2 links2 sources
Open the reference →
96

Tree Search

Tree Search for Language-Model Reasoning

Scaling & Alignment2 links4 sources
Open the reference →
97

Video World Models

Video World Models

Generative Models2 links3 sources
Open the reference →
98

Self-Improvement

Self-Improvement & Distillation Loops

Scaling & Alignment2 links3 sources
Open the reference →
99

Model Collapse

Model Collapse

Theory2 links3 sources
Open the reference →
100

Infinite Context

Infinite-Context Architectures

Efficiency2 links4 sources
Open the reference →

About These Pages

What each page contains

A short summary, questions or points that are easy to miss, the key equation and the math behind it, key papers and links to related concepts.

Interactive demos

40 of the 100 pages include an interactive demo. Use the Demos filter above to list them.

Newer notebooks

Some concepts also have a newer notebook with its own demo. Those pages link to it at the top.