Speaker Notes

Jeff Dean & Noam Shazeer — 25 years at Google: from PageRank to AGI

Jeff Dean and Noam Shazeer trace 25 years of scaling machine learning at Google, detailing the shift from hardware-constrained heuristics to massive deep learning, custom TPU hardware, and future architectural paradigms.

Dwarkesh Patel · 2:15:36 · YouTube ↗

Podcast episode · 1 host and 2 guests  Advanced · Hardware and scaling strategy

Highlights

15 clips · 24 min of 136 min
    • Arithmetic has become extremely cheap while moving data remains expensive.
    • Deep learning scaled because matrix multiplications maximize arithmetic intensity.
    • TPUs were built as reduced-precision linear algebra engines to exploit this dynamic.
    • Inference and training precision have dropped from 8-bit integers to INT4 and even single-bit representations.
    • Algorithm designers naturally resist quantization due to perceived risk and irritation.
    • Co-designing chips and algorithms forces adoption when throughput increases multi-fold.
    • An initial translation system took 12 hours per sentence due to 100,000 disk seeks.
    • Designing an in-memory compressed 5-gram representation across 200 machines solved the bottleneck.
    • Batched APIs allowed parallel lookups that dropped latency to 100 milliseconds.
    • Training a neural net on 10 million YouTube frames across 16,000 CPU cores used unsupervised learning.
    • Without explicit labels, neurons naturally emerged that responded specifically to cat faces.
    • This 50x larger scale advanced ImageNet state-of-the-art by 60% relative improvement.
    • Naive attention mechanisms scale quadratically, preventing direct scaling to trillions of tokens.
    • Context windows provide sharper, more precise data recall compared to squishy parameter-based memorization.
    • Language model generation is drastically cheaper than human services like customer support or professional labor.
    • Scaling compute at inference time provides an immediate economic path to making models substantially smarter.
    • Small drafter models propose multiple potential tokens in parallel for the larger model to verify.
    • This generation-and-verification pattern successfully bypasses the single-token decode bottleneck in transformers.
    • Small models generate candidate token sequences while large models verify batches in parallel.
    • This technique significantly increases inference throughput without sacrificing output quality.
    • Using the big model as a verifier shifts the compute pattern from sequential decoding to parallel verification.
    • Asynchronous training scales easily on slow hardware but yields non-reproducible results tied to background machine noise.
    • High-bandwidth network connections across metro areas make multi-data center synchronous training viable.
    • Synchronous training provides a clean mental model and replicable experiments for researchers.
    • Monolithic pre-training forces rigid trade-offs between different capabilities like coding and multilingual data.
    • Sparse, modular models allow independent teams to develop and plug in specialized sub-components.
    • Decoupled components make research parallelizable and enable targeted capability upgrades.
    • Modular model architectures allow freezing a baseline and testing variant sub-modules.
    • Targeted module updates make research significantly cheaper than retraining from scratch.
    • Decoupling development enables parallelized improvements across different model capabilities.
    • Distillation can happen continuously at the individual module level in the background.
    • Small distilled versions handle simple tasks while large versions capture complex capabilities.
    • Once distillation finishes, parameter capacity can be freed up to learn new information.
    • MoE models do not save memory by skipping unused weights during high-batch inference.
    • Batching independent requests means all experts must remain loaded in HBM simultaneously.
    • Hardware pods with massive in-memory capacity are essential for serving these architectures.
    • Model connectivity should organically mirror the physical hierarchy of the underlying hardware.
    • Dense connections belong on-chip, while inter-chip links should be sparse and bottlenecked.
    • Allowing connectivity to emerge organically avoids guessing the right architectural proportions.
    • Standard next-token prediction may not extract maximal value from every training token.
    • Masking or hiding information forces models to reason harder, mirroring vision techniques.
    • Increasing training difficulty can yield greater capability without exhausting text data.
    • Search engines demand high factuality, making hallucination-prone early models seem unready.
    • Safety concerns and strict reliability bars initially slowed down public deployments.
    • Teams underestimated how much users would value open-ended chat assistance over strict search.

Questions this video answers

15 · every answer cites the moment it was said
01 How did the scaling of hardware and arithmetic change the design of deep learning systems? 2 sources · 7:50

Arithmetic became extremely cheap while moving data around became relatively expensive, prompting a pivot toward hardware specialized for matrix multiplications. Companies like Google started building TPUs designed as reduced-precision linear algebra machines to maximize throughput. This hardware-algorithm co-design unlocked the massive scale needed for modern deep learning.

02 Why did the early 2007 N-gram machine translation model require a custom in-memory data structure? 2 sources · 17:32

The initial machine translation model won translation contests with high quality, but took 12 hours per sentence because it relied on 100,000 disk seeks per word. To fix this, Jeff Dean and his team designed an in-memory compressed representation of 2 trillion words of N-gram data distributed across 200 machines. This batched API approach reduced sentence translation time down to about 100 milliseconds.

03 How did unsupervised learning on YouTube frames advance neural network training in the early days of the Brain team? 2 sources · 25:47

The team built infrastructure to train large neural nets across thousands of CPU cores using unsupervised learning on 10 million randomly selected YouTube frames. Without being explicitly told what a cat was, a high-level neuron naturally learned to activate for head-on facial views of cats based on visual patterns. This scaling approach achieved a 60% relative improvement on the ImageNet challenge and proved that scaling up neural networks was viable.

04 What are the memory and computational trade-offs of long-context models versus model parameters? 2 sources · 33:10

While model parameters efficiently memorize facts, long-context models provide sharp, clear information via transformer attention mechanisms. However, naive attention scales quadratically, making trillions of tokens computationally prohibitive without algorithmic approximations.

05 How is AI currently impacting software developer productivity at Google? 2 sources · 36:49

Google has further trained internal Gemini models on their proprietary codebase to assist developers. Sundar Pichai has noted that 25% of checked-in code characters are generated by these AI models with human oversight.

06 Why is inference-time compute becoming a major area for model improvement? 2 sources · 53:56

Language model inference is significantly cheaper than human labor or alternative pastimes, leaving vast economic headroom to trade compute for smarter answers. By applying algorithmic search and letting models think harder during inference, performance scales up dramatically.

07 How is multi-data center training managed across multiple metro areas? 2 sources · 1:03:27

Large models can be trained across multiple data centers using high-bandwidth, high-latency connections between metro areas. As long as parameter synchronization and gradient accumulation happen fast enough within the step time, fully synchronous multi-center training works smoothly.

08 How does Google handle the integration of multiple research improvements at scale? 2 sources · 1:08:55

Researchers first invent improvements in isolation using small, simple codebases with fast turnaround times. Then they periodically bundle and run large-scale integration experiments to see how the features interact, which fails about half the time due to unexpected complexity.

09 What role can language models play in ensuring AI safety and control? 2 sources · 1:19:06

Analyzing text is generally easier than generating it, allowing models to inspect their own and other systems' outputs for dangerous or problematic content. Additionally, exposing model capabilities through controlled APIs and user interfaces provides an extra layer of governance.

10 How could modular model structures enable continual learning? 2 sources · 1:37:09

Instead of monolithic pre-training, modular architectures like mixture-of-experts could feature independent components developed by different teams globally. Continuous distillation can then update these smaller sub-modules over time without requiring the entire model to be retrained from scratch.

11 How can we scale models modularly using distillation? 2 sources · 1:41:45

Distillation can be applied at the module level in a continual process where a large model continually distills into a smaller variant. Once trained, the smaller module replaces the large one, freeing up parameter capacity for new data. This allows independent teams to update specific capabilities like coding or languages without retraining from scratch.

12 Do Mixture of Experts models save memory by skipping unused weights? 2 sources · 1:45:28

No, Mixture of Experts models still require the entire model to be loaded into memory to achieve high inference efficiency at large batch sizes. At inference, you send a fraction of a large batch through every expert rather than skipping weights entirely. Therefore, you need enough High Bandwidth Memory to fit the whole model.

13 How should hardware topology influence neural network connectivity? 2 sources · 1:54:46

Model connectivity should mimic the underlying hardware topology to optimize communication costs. Networks should have extremely dense connections within the same chip and HBM, scaling down to fewer bottlenecked connections across chips and TPU pods. Allowing this structure to emerge organically prevents guessing the wrong connection proportions.

14 How can we improve training sample efficiency beyond next-token prediction? 2 sources · 1:58:13

Training objectives can be modified to force models to work harder on every token, similar to vision model techniques like dropout or hiding partial information. Additionally, allowing models to take actions and observe consequences—much like humans and infants do—would dramatically improve sample efficiency. Exploring self-play and thought experiments can also generate high-value learning without external data.

15 Why did Google delay releasing early chatbots like Meena? 2 sources · 2:08:34

Google initially held back chatbot releases because search engine standards demand high factuality and zero hallucinations, which early models failed to meet. They also needed to address safety issues regarding offensive outputs. They underestimated how much users would value chatbots for general utility tasks rather than strict search queries.

Section-by-section summary

26 parts

Growing from a tiny team to managing massive scale

As companies expand from dozens of employees to thousands, founders and early engineers inevitably lose track of individual names and projects. Maintaining a strong internal network with a single level of indirection remains essential for finding the right experts across a large organization.

Joining early Google with ambitions for AI research

Early employees joined the search startup to gain financial stability before eventually funding their own artificial intelligence research. Google's broad mandate to organize the world's information naturally aligned with the long-term pursuit of advanced AI.

Hardware-driven algorithms and extreme quantization

Slowing CPU scaling shifted the industry toward specialized ML accelerators like TPUs optimized for low-precision arithmetic. Co-designing chips and algorithms to use extreme quantization like INT4 or lower drastically increases throughput-to-cost ratios.

Scaling N-gram models to solve machine translation

Training a massive two-trillion token N-gram model with an in-memory compressed data structure reduced machine translation latency from 12 hours to 100 milliseconds. This early engineering breakthrough demonstrated how web-scale statistics could solve complex language tasks.

Self-supervised learning as an AI-complete foundation

Unsupervised language modeling provides an endless supply of training data by predicting missing words from surrounding context. This self-supervised approach serves as an AI-complete foundation capable of generalizing to spelling correction, code completion, and beyond.

Scaling up neural networks for unsupervised feature learning

Training a massive neural network across thousands of CPU cores on YouTube frames enabled unsupervised discovery of high-level concepts like cat faces. Achieving a massive relative improvement on ImageNet proved that scaling up neural network compute was a winning strategy.

Expanding from information retrieval to multimodal creation

Organizing the world's information extends far beyond traditional search retrieval into multimodal understanding and content generation. Modern AI systems can process diverse inputs like code, video, and audio to actively perform complex work for users.

The memory trade-off in long-context models

While model parameters efficiently memorize facts, context windows offer sharp, precise attention over millions of tokens. Scaling attention to trillions of tokens requires massive algorithmic approximations due to the quadratic complexity of naive attention.

AI-generated code boosting developer productivity

Training models on internal code bases yields significant productivity gains, with a substantial portion of recent code check-ins generated by AI. Future software engineering tools will allow developers to delegate entire architectural specs and explore experimental research ideas automatically.

Algorithmic progress driven by automated exploration

Model improvements come as much from algorithmic and architectural breakthroughs as from raw hardware scaling. Automated exploration of ideas allows researchers to vet thousands of small-scale concepts before committing them to production training runs.

Accelerating chip design loops with AI

Traditional chip design cycles take 18 months, but automated search spaces and AI assistance can shrink this timeline drastically. Shorter hardware design cycles allow teams to build specialized chips tailored closely to upcoming machine learning algorithms.

Unlocking intelligence via inference-time compute

Language model generation is dramatically cheaper than human labor, providing massive headroom to scale compute at inference time. Allowing models to think harder, search, and iteratively explore solutions yields disproportionately better answers for complex tasks.

Optimizing inference efficiency with drafter models

Inference workloads will increasingly split into latency-sensitive user requests and asynchronous background tasks. Techniques like using small drafter models to propose multiple tokens for a large model to verify help bypass the single-token decode bottleneck.

Managing multi-data center training and scale

Large models can be trained synchronously across multiple metro areas by leveraging high-bandwidth connections relative to step times. While neural networks are remarkably tolerant of training noise, scaling up exposes hidden bugs that complicate debugging.

Integrating experimental improvements requires careful scaling

Researchers test ideas in isolation at small scale, but combining them at large scale fails about half the time due to unexpected interactions. Keeping the codebase clean and managing algorithmic complexity is essential to handle these trade-offs.

AI safety and steering become paramount

As models rapidly improve and approach recursive self-improvement, the risk of misuse or misalignment grows significantly. Engineering safe systems and using models to analyze and verify outputs are critical to capturing benefits safely.

Physical workspaces foster rapid developer collaboration

Early Google search architecture and modern Gemini development share a reliance on vibrant, in-person environments like micro-kitchens. These shared physical spaces and chat rooms accelerate idea cross-pollination and collaborative problem-solving.

Inference demand will drive massive compute growth

Scaling inference compute—where complex tasks require thousands of times more tokens and global adoption approaches 100%—will require an astronomical increase in energy and hardware. Co-designing hardware and models is vital to make this accessible.

Modular models enable continual learning

Current monolithic models make incremental updates difficult, but sparse, modular architectures could allow independent teams to upgrade specific capabilities. This decouples model components like languages or coding logic for faster, parallel research.

Continuous distillation powers module-level scaling

Distillation can operate at the individual module level rather than just for final model compression. Running background processes to continually distill large modules into smaller, faster versions frees up parameter capacity for new learning. Furthermore, experts within these models remain relatively interpretable using sparse autoencoders.

Mixture of experts requires full model in memory

Mixture of experts does not mean unused weights can be omitted from memory during high-batch inference. Efficient serving relies on processing large, independent batch sizes across all experts simultaneously. Consequently, data centers must host the entire model footprint within high-bandwidth memory.

Hardware topology should dictate model connectivity

Artificial neural networks can draw deeper inspiration from biological brains by organizing into specialized, organically grown regions. Rather than hand-specifying architecture, model connectivity should organically mirror the physical hardware hierarchy. Dense connections belong on-chip, while sparse, bottlenecked connections should handle inter-chip and cross-datacenter communication.

Pre-training objectives need greater data efficiency

Standard next-token prediction may not extract maximal value from training data compared to how humans learn. Introducing tougher training objectives, such as masking or forcing models to work harder on complex tokens, can drastically improve sample efficiency. Exploring actions in simulated environments also bridges the gap between passive observation and active reasoning.

Balancing open research with product integration

Releasing foundational breakthroughs like the Transformer has massively accelerated the broader AI ecosystem. While open publishing fosters community progress, companies must balance it by hardening critical capabilities inside commercial products first. Different developments warrant different disclosure strategies depending on their strategic value.

Search constraints delayed early chatbot releases

Google possessed capable internal chatbots early on, but search-driven standards for factuality and strict safety bars caused hesitation. Because search demands high precision, the team initially underestimated how much users would value chat models for open-ended assistance. Over time, refining these capabilities led to robust deployment via the Gemini product line.

Curiosity and flexibility drive career longevity

Sustaining multi-decade technical breakthroughs requires diving into unfamiliar domains and collaborating with multidisciplinary teams. Cultivating intellectual humility allows researchers to abandon failing ideas quickly rather than clinging to sunk costs. Balancing top-down coordination with bottom-up experimentation creates an environment primed for continuous innovation.

0:00 / 2:15:36
k play/pause · f fullscreen · m mute
n next · p previous clip · t highlights/full
< > slower/faster · / search what was said
  • TPU10× · first 7:22

    Specialized machine learning hardware designed by Google as reduced-precision linear algebra machines.

  • N-gram3× · first 13:43

    Statistics for how often every sequence of N words occurs in a large text corpus.

  • Transformer9× · first 1:19

    A neural network architecture utilizing attention mechanisms that revolutionized machine learning scaling.

  • Drafter models1× · first 1:01:46

    Smaller language models used to predict multiple tokens ahead to accelerate big model inference efficiency.

  • TPU pods3× · first 1:04:50

    High-performance TPU hardware clusters with incredible interconnect bandwidth used for synchronous training.

  • Pathways3× · first 1:39:53

    Infrastructure framework supporting complex, twisty model architectures and asynchronous updates.

  • Mixture of Experts6× · first 1:24

    A neural network architecture that routes different inputs to specialized subnetworks or experts.

  • Sparse Autoencoder1× · first 1:43:59

    A technique used for model interpretability to deduce the specific characteristics of individual neurons.