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 Implementation to Auto-TuningBlock Pointers and Multi-Dim SupportCausal Masking OptimizationGrouped Query AttentionBackward Pass Implementation
Distributed Training
Data ParallelismZeRO OptimizerFully Sharded Data ParallelTensor ParallelismPipeline ParallelismMulti-Dimensional Hybrid Parallelism

Hands-on Training

Overview
Pretraining
Pretraining DataTokenizer TrainingModel ArchitectureData PipelineTraining LoopMonitoring and Validation
X (Twitter)
FundamentalsModel ArchitecturePosition Encoding

Position Encoding Basics

Premium

Why Transformers need position information, and the methods and limits of absolute position encoding

Get code access

Permutation Invariance in Transformers

Let’s start with a key fact: standard Self-Attention does not care about input order.

Recall the Attention formula:

Log in to continue reading

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

Attention(Q,K,V)=softmax(QKTd)V\text{Attention}(Q, K, V) = \text{softmax}\left(\frac{QK^T}{\sqrt{d}}\right)VAttention(Q,K,V)=softmax(d

Here Q=XWQQ = XW_QQ=XWQ​, K=XWKK = XW_KK=XWK​, V=XWVV = XW_V. If we shuffle the tokens in , the , , and vectors for each token do not change, only their positions in the sequence do. Attention computes pairwise dot products between all tokens, which is .

In other words, “I like cats” and “Cats like me” look the same to Attention. That is clearly unacceptable because language meaning depends on word order.

That is why we need position encoding: inject position information into the input so the model knows where each token is in the sequence.

Rotary Position Embedding

From position encoding basics to RoPE math, implementation, and length extrapolation

RoPE Math Derivation

From complex rotations to higher-dimensional generalization, understand the core math of rotary position embeddings

Table of Contents

Permutation Invariance in Transformers
Absolute Position Encoding
Sinusoidal Position Encoding
Why sin/cos?
Learned Position Encoding
How Absolute Position Encoding Is Used
Hidden Properties of Sinusoidal PE
Dot Product Depends Only on Relative Position
Long-Range Decay
Limits of Absolute Position Encoding
Summary
​
QKT
​
)
V
V
=
XWV​
XXX
QQQ
KKK
VVV
order-invariant