The Transformer (Vaswani et al., 2017) replaced RNN/LSTM by processing entire sequences in parallel, unlocking GPU scaling. Foundation of every modern LLM.
+------------------ INPUT ------------------+
| tokens: [BOS, The, cat, sat, EOS] (ids) |
+------------------------------------------+
|
v
+-------------------------------+
| Token Embedding + Positional |
+-------------------------------+
|
v
+---------------------------------------+
| N x (Multi-Head Self-Attention -> |
| LayerNorm -> FeedForward -> |
| LayerNorm) |
+---------------------------------------+
|
v
+-----------------------+
| Linear -> Softmax |
| over vocabulary |
+-----------------------+
|
v
next-token probability distribution
A stack of identical blocks (attention + feed-forward) repeated N times. Each block is parallel over sequence and sequential over depth. The original 2017 paper had encoder + decoder halves; modern frontier LLMs (GPT-4, Llama 3, Claude, Mistral) are decoder-only — the recipe is identical, they just skip the encoder half.
def block(x):
a = layer_norm(x + self_attention(x)) # attention sublayer
out = layer_norm(a + feed_forward(a)) # FFN sublayer
return out
A full model is for _ in range(N): x = block(x) followed by a softmax head.
O(L^2) attention). Mitigations: FlashAttention, sliding window, multi-query / grouped-query attention.Reading a whole paragraph at once and scribbling notes about which words relate to which, instead of word-by-word.
Interview tip: “How would you scale an RNN?” — answer: use a Transformer.