Skip to content

Multi-head attention section - #132

Open
topepo wants to merge 2 commits into
mainfrom
MHA
Open

topepo wants to merge 2 commits into
mainfrom
MHA

Conversation

@topepo

@topepo topepo commented Sep 6, 2026

Copy link
Copy Markdown
Contributor

Preview at dev.aml4td.org


2. **Column self-attention**: 3 blocks, 2 heads of dimension 13, block output dimension (2 heads × 13 dimensions = 26)
- Block 1: input dimension 7 to output dimension 26, yielding output shape (16 × 26)
- QKV projections: (3 × (2 × 7 × 13)) = 546 parameters

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I'm not sure whether to include the number of parameters for each item or just the total for that section.

When we get to the foundational model section, the analogous diagram gets pretty hard to read.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant