#776 · Primary category: Education & Research
pytorch-original-transformer
My implementation of the original transformer model (Vaswani et al.). I've additionally included the playground.py file for visualizing otherwise seemingly hard concepts. Currently included IWSLT pretrained models.
Project last updated:12/27/20
GitHub Stars
1.1K
Forks
188
Contributors
1
License
MIT
Why we included this project
This repo is a readable PyTorch port of the original transformer architecture from Vaswani et al., written for learning rather than for chasing leaderboard numbers. The code is heavily commented, so each component is explained in place instead of arriving as an opaque library call. The playground.py script is the part worth checking first: it visualizes positional encodings and the custom learning-rate schedule, ideas that are easy to describe in prose but hard to picture from the formulas alone. Pretrained IWSLT translation models and a companion notebook give you something real to run once the theory starts to click. For students or self-taught engineers trying to understand the ideas behind modern LLMs, this is a friendly place to build that intuition before moving on to higher-level frameworks.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
prompts.chat
f.k.a. Awesome ChatGPT Prompts. Share, discover, and collect prompts from the community. Free and open source — self-host for your organization with complete privacy.
JavaGuide
Java Interview & Backend General Interview Guide, covering computer fundamentals, databases, distributed systems, high concurrency, system design, and AI application development.
system-prompts-and-models-of-ai-tools
A curated collection of system prompts, internal tools, and AI models from popular AI assistants and coding agents.
30-seconds-of-code
Coding articles to level up your development skills
generative-ai-for-beginners
21 Lessons, Get Started Building with Generative AI