Color Key
Input / output
Embedding layers
Self-attention
Feed-forward network
Normalization
Layer Details
Hover over a layer to see what it does and its tensor dimensions.