A numerical stable implementation of Transformer from scratch using PyTorch. Features Pre-Norm architecture analysis and manual Attention mask fixes.
-
Updated
Dec 10, 2025 - Python
A numerical stable implementation of Transformer from scratch using PyTorch. Features Pre-Norm architecture analysis and manual Attention mask fixes.
Add a description, image, and links to the pre-norm topic page so that developers can more easily learn about it.
To associate your repository with the pre-norm topic, visit your repo's landing page and select "manage topics."