Learn how the attention mechanism in LLMs helps convert base token embeddings into rich context-aware embeddings.
Short CourseBeginner1 hour 6 mins
Attention in Transformers: Concepts and Code in PyTorch
Instructor: Josh Starmer
- Beginner
- 1 hour 6 mins
- 11 Video Lessons
- 3 Code Examples
- Instructor: Josh Starmer
What you'll learn
Understand the Query, Key, and Value matrices, what they are for, how to produce them, and how to use them in attention.
Learn the difference between self-attention, masked self-attention, and cross-attention, and how multi-head attention scales the algorithm.
About this course
This course clearly explains the ideas behind the attention mechanism. It walks through the algorithm itself and how to code it in Pytorch. Attention in Transformers: Concepts and Code in PyTorch, was built in collaboration with StatQuest, and taught by its Founder and CEO, Josh Starmer.
The attention mechanism was a breakthrough that led to transformers, the architecture powering large language models like ChatGPT. Transformers, introduced in the 2017 paper “Attention is All You Need” by Ashish Vaswani and others, revolutionized AI with their scalable design.
Learn how this foundational architecture works to improve your intuition about building reliable, functional, and scalable AI applications.
What you’ll do:
- Understand the evolution of the attention mechanism, a key breakthrough that led to transformers.
- Learn the relationships between word embeddings, positional embeddings, and attention.
- Learn about the Query, Key, and Value matrices, how to produce them, and how to use them in attention.
- Go through the math required to calculate self-attention and masked self-attention to learn how and why the equation works the way it does.
- Understand the difference between self-attention and masked self-attention, and how one is used in the encoder to build context-aware embeddings and the other is used in the decoder for generative outputs.
- Learn the details of the encoder-decoder architecture, cross-attention, and multi-head attention, and how they are incorporated into a transformer.
- Use PyTorch to code a class that implements self-attention, masked self-attention, and multi-head attention.
Who should join?
Anyone who has basic Python knowledge and wants to learn how the attention mechanism in LLMs like ChatGPT works.
Course Outline
11 Lessons・3 Code ExamplesIntroduction
Video・6 minsThe Main Ideas Behind Transformers and Attention
Video・4 minsThe Matrix Math for Calculating Self-Attention
Video・11 minsCoding Self-Attention in PyTorch
Video with Code Example・8 minsSelf-Attention vs Masked Self-Attention
Video・14 minsThe Matrix Math for Calculating Masked Self-Attention
Video・3 minsCoding Masked Self-Attention in PyTorch
Video with Code Example・5 minsEncoder-Decoder Attention
Video・4 minsMulti-Head Attention
Video・2 minsCoding Encoder-Decoder Attention and Multi-Head Attention in PyTorch
Video with Code Example・4 minsConclusion
Video・1 min
Instructor
Course access is free for a limited time during the DeepLearning.AI learning platform beta!
Want to learn more about Generative AI?
Keep learning with updates on curated AI news, courses, and events, as well as Andrew’s thoughts from DeepLearning.AI!