MODULE 1  ยท  How LLMs Work

Transformers: The Paper That Changed Everything

Have you ever wondered โ€” how does ChatGPT actually understand what you're saying? The answer is one word: Attention. And it all started with one 15-page paper.

๐Ÿ“… Mar 2026
โฑ 10 min read
๐ŸŽฏ Episode 2 of 98
TransformersSelf-AttentionArchitectureAI Basics
In this episode

You type a sentence. It responds with something that makes sense. But how? How does a machine figure out that when you say "The cat sat on the mat because IT was tired" โ€” the "IT" refers to the cat, not the mat?

The answer is Attention. And it all started with one paper.

The Paper That Started It All

In 2017, eight Google engineers published a paper called "Attention Is All You Need."

15 pages. That's it.

From those 15 pages came:

๐Ÿค– ChatGPT (OpenAI)

๐Ÿ”ต Claude (Anthropic)

๐Ÿ’Ž Gemini (Google)

๐Ÿฆ™ Llama (Meta)

โšก

Mistral, Command R, Grok โ€” every major AI model

The architecture they introduced? The Transformer.

Before transformers, AI processed language one word at a time โ€” sequentially. Read the first word, then the second, then the third. Like reading through a straw. Slow, and by the time you reached the end of a sentence, you'd half-forgotten the beginning.

The big idea

Transformers flipped this completely. They said: look at ALL words at once. In parallel. That single idea changed everything.

The Core Idea: Self-Attention

In Episode 1, we learned that AI doesn't read words โ€” it reads tokens. Today's question is: once the model has those tokens, how does it understand them?

The answer is self-attention โ€” every token looks at every other token and asks: "How relevant are you to me?"

The Classic Example

Consider this sentence:

"The cat sat on the mat because IT was tired."

"IT" refers to what? The cat. Not the mat. You figured that out in one second. But a computer needs a mechanism to make that connection. That mechanism is self-attention.

Here's what happens:

The model looks at the token "IT"

It compares "IT" with every other token in the sentence

It computes an attention score for each pair โ€” how strongly are these two tokens related?

"IT" โ†” "cat" โ†’ high score (strongly related)

"IT" โ†” "mat" โ†’ low score (weakly related)

"IT" โ†” "tired" โ†’ high score (IT was tired)

The model doesn't just do this for "IT" โ€” it does this for every single token, all at once, in parallel. Every word is simultaneously checking its relationship with every other word.

Try It Yourself โ€” Interactive Demo

Here's a live visualization of self-attention. Click any word to see its attention scores:

Self-Attention Visualizer

Click any word to see its attention scores

Click any word to see which other words it pays attention to

Strong

Medium

Weak

How Attention Actually Works (The Math)

Don't worry โ€” this is simpler than it sounds.

Every token in the sentence gets converted into three vectors:

snippet
code
Query (Q) โ€” "What am I looking for?"
Key (K) โ€” "What do I contain?"
Value (V) โ€” "What's my actual information?"

Step 1: Compute Scores

Take the Query of one token and multiply it (dot product) with the Key of every other token.

snippet
code
Score("IT", "cat")   = Query_IT ยท Key_cat   = high number
Score("IT", "mat")   = Query_IT ยท Key_mat   = low number
Score("IT", "tired") = Query_IT ยท Key_tired = high number

The dot product measures similarity. If two vectors point in the same direction โ†’ high score โ†’ these tokens are related.

Step 2: Scale

Divide all scores by โˆšd (square root of the dimension). This keeps the numbers from getting too large and destabilizing training.

snippet
code
Scaled_Score = Score / โˆšd

// For GPT-4o with dimension 128, that's dividing by ~11.3

Step 3: Softmax

Convert all scores into probabilities that add up to 1.0:

Attention weights for "IT":

โ†’ cat: 0.42 (strongest โ€” IT refers to cat)

โ†’ tired: 0.28 (IT was tired)

โ†’ was: 0.18

โ†’ because: 0.06

โ†’ mat: 0.03 (weak โ€” IT doesn't refer to mat)

โ†’ sat: 0.02

โ†’ on: 0.01

Step 4: Weighted Sum

Multiply each token's Value vector by its attention weight, then sum them all up. This gives "IT" a new representation that contains information about the cat being tired.

snippet
code
New_IT = 0.42 ร— Value_cat + 0.28 ร— Value_tired + 0.18 ร— Value_was + ...

This is the magic

After self-attention, the token "IT" doesn't just represent the word "it" anymore โ€” it carries context from the entire sentence. It knows it refers to the cat. Nobody programs these Q, K, V vectors. The model learns them during training by seeing billions of sentences. It figures out on its own that "IT" and "cat" should have high attention when they're in a coreference relationship.

Multi-Head Attention: Multiple Perspectives at Once

Real transformers don't just run attention once. They run it multiple times in parallel โ€” these are called attention heads.

Why? Because a single attention pass can only capture one type of relationship. But language has many:

Head 1 might learn: syntactic relationships (subject โ†” verb)

Head 2 might learn: coreference ("IT" โ†” "cat")

Head 3 might learn: positional relationships (nearby words)

Head 4 might learn: semantic similarity

GPT-4o: 128 heads per layer

GPT-4o reportedly uses 128 attention heads per layer โ€” 128 different "perspectives" on every token relationship, computed simultaneously. The outputs of all heads are concatenated and projected back into a single vector. This gives the model a rich, multi-faceted understanding of each token's relationship to every other token.

Why Transformers Are Fast: Parallelism

Before transformers, the dominant architectures were RNNs (Recurrent Neural Networks) and LSTMs.

These processed tokens sequentially โ€” one at a time, left to right:

RNN / LSTM (old way)

snippet
code
Token 1 โ†’ process โ†’ Token 2 โ†’ process โ†’ Token 3 โ†’ process โ†’ ...

Like reading a book one letter at a time. By the time you reach word 1,000, you've nearly forgotten word 1.

Transformers process all tokens simultaneously:

snippet
code
Transformer (modern)
Token 1 โŸฉ
Token 2 โŸฉ โ†’ process ALL AT ONCE โ†’ output
Token 3 โŸฉ

... โŸฉ

Why GPUs love transformers

GPUs are designed for parallel computation โ€” thousands of calculations at the same time. RNNs couldn't fully utilize GPUs because each step depended on the previous one. Transformers have no such dependency. Result: Training that would have taken months with RNNs takes weeks with transformers. Inference that would take seconds takes milliseconds.

The Full Transformer Architecture (Simplified)

A transformer layer does two main things:

Self-Attention โ€” every token looks at every other token (what we just learned)

Feed-Forward Network โ€” each token passes through a neural network independently (adds computational depth)

These two steps repeat many times โ€” stacked in layers:

Input tokens

example
code
โ†“
snippet
code
[Self-Attention โ†’ Feed-Forward] โ† Layer 1
โ†“
[Self-Attention โ†’ Feed-Forward] โ† Layer 2
โ†“
[Self-Attention โ†’ Feed-Forward] โ† Layer 3
โ†“
    ... (GPT-4o has ~120 layers)
    โ†“

Output predictions

Layers build meaning progressively

Each layer builds a deeper understanding. Early layers might learn basic grammar. Middle layers learn meaning. Later layers learn complex reasoning and nuance.

The Numbers

Model Layers Attention Heads Parameters

GPT-2 48 25 1.5B

GPT-3 96 96 175B

Llama 3 70B806470B
GPT-4o~120 (est.)~128 (est.)Unknown

More layers and heads = deeper understanding, but also more compute and cost. This is the tradeoff we'll explore in Episode 3 when we talk about parameters.

Why This Matters for You

Understanding transformers isn't just academic. It directly impacts how you use AI:

  1. Long prompts lose context

Self-attention has O(nยฒ) complexity. Double the tokens, quadruple the compute. This is why there's a context window limit.

  1. Order matters less than you think

Unlike RNNs, transformers see all tokens at once. Putting important instructions at the start vs. end matters less (though models do have position biases).

  1. Prompt engineering is attention engineering

When you write a clear, specific prompt, you're helping the model's attention focus on what matters. Vague prompts = scattered attention = worse outputs.

  1. Why AI "hallucinates"

The model generates tokens based on attention patterns learned from training data. If it hasn't seen enough examples of a topic, the attention patterns are weak, and it fills gaps with plausible-sounding but incorrect tokens.

The hallucination root cause

Hallucination isn't a bug โ€” it's a fundamental property of how attention-based generation works. The model doesn't "know" things. It predicts tokens based on patterns. When patterns are weak, it guesses. This is why grounding (RAG, tools, citations) matters so much in production.

โ† Previous Ep 1: Tokens: What AI Actually Sees (Not Words)