Google's AI Infrastructure Chief, Amin Vahdat, on the Physics & Economics of Frontier AI
At 100,000-accelerator scale, something fails multiple times an hour, which is why Google's Chief Technologist for AI Infrastructure Amin Vahdat thinks FLOPS is a vanity metric. The metric that matters is what he calls goodput: the useful work a workload actually delivers through real failures. Amin walks through the calculus that split the TPU line into…
People
Notes
-
What Is an AI Data Center? at 1:55
-
-
-
-
-
-
-
-
-
-
-
-
-
Related notes
- What Is the Key Question for AI Progress?
- How Much Electricity Does a 100 Gigawatt AI Cluster Use?
- Why Can't Chip Fabs 10x Wafer Output for AI?
- Why Would AI Data Centers Be Cheaper in Space?
- Why Does Google Want Data Centers in Space?
- Why Is Google Putting AI Data Centers in Space?
- Will AI Data Centers Raise Electric Bills?
- Who Carries the Debt on AI Data Center Leases?
Full transcript
The episode notes above make this conversation easy to skim.
0:00
In hardware, again, as you know, there is this opportunity where the more you specialize to a particular workload, the less flexible it is, the faster, the more power efficient the hardware is going to be. So it is this art and it's this projection of what are you designing to and how persistent is that workload? In other words, if it's going to go away after a month or two months or three months, even if it's big for those three months, you got a really narrow window to intercept it.
0:28
So it has to be somewhat durable. And you have to be able to project exactly what win you can get for specializing to it. Thrilled to welcome Amin Vedat to the show. Amin, thank you for joining us for today. It's really exciting to be here. I'm really looking forward to it.
1:00
I am very excited for today's topic because we are in the middle of the biggest CapEx build out in human history. You are in the middle of it. Google alone is expected to spend more than $200 billion on CapEx this year, most of which is going into building data centers. You're at the center of it all. You were named the head of Google's AI infra at the end of last year.
1:22
So you are the man spearheading the efforts of one of the most capital intensive build outs in human history. And so I am really excited to get into it with you today. It's a huge year for sure, of course, across the industry, including at Google. Really, I don't think we've ever seen anything like this, honestly. Certainly not at Google, but as you said, I think in the history of humanity in terms of the build outs and the pace of transformation, really unbelievable.
1:47
Before we get into it, maybe just crash course for the audience. What is an AI data center and how is it different from a non-AI data center? It's a very good question in that AI data center and a non-AI data center do have a lot of similarities actually, and they're not radically different. I mean, it consists of concrete, an enclosure. It consists of electrical yards, mechanical yards, cooling.
2:12
You have row after row of power being distributed. There's a huge amount of network infrastructure there. In other words, we connect large amounts of computes to one another. A fair amount of storage infrastructure goes in there. I think that the big difference that we're seeing with AI infrastructure is specialization. In the past, when we're building a data center, it really is 20, 25, 30 year building investments.
2:38
And we're thinking about how it's going to be evolving over that 20, 25, 30 year period. You might have servers go into it. You might have networking storage. You might have some accelerators, GPUs, TPUs, whatever they might be going into it. But it's a 25, 30-year planning horizon. And the lifetime of the hardware might be six years. So many generations that we have to plan for.
3:03
An AI data center oftentimes is going to be more purpose-built. So in other words, we are oftentimes
3:12
co-designing the building with the hardware that might go into it. We might be saying, you know what, actually, we're not going to put a lot of storage into this building. Why? Because a storage rack might have 10, 20, 30, 40 kilowatts of power. You put that next to a TPU rack or a GPU rack, that easily is hitting hundreds of kilowatts today.
3:34
It might hit megawatts. People are talking about that in the next few years. Designing a building that might take 30 storage racks in a row versus one or two AI racks in that same row, very, very different design. Just think of it in terms of the size. Think of it in terms of the power and how that would be distributed across the building, et cetera.
3:56
Networking would be the same. The amount of networking that you would need for a storage rack, tiny, especially if it's hard drives, compared to an AI rack. If you're trying to make it fully fungible, you'll probably make it too big and too overbuilt, fungible over certainly a 30-year period. an AI data center likely is going to be much more purpose-built, co-designed with the hardware, even to the point of cooling and power distribution, et cetera.
4:25
So a lot more co-optimization. You all delivered a big Vera Rubin cluster to one of my portfolio companies, Ineffable Intelligence, and I saw the photographs of it as it went out. That is a thing of beauty. It is. It is. It's one of those things where, similar to looking at the biggest construction projects in human history, you look at that data center and it's like, wow, that is a monument to what mankind can do.
4:49
Now, we got a nice picture of it, and this is just a couple of racks. The cabling, the fiber distribution for it is really beautiful. I mean, beauty is not the beholder, but for folks like me and probably yourself and others in the audience as well, it is a thing of beauty. We put this picture up on social media. And people love the fact that Ineffable, they're a huge fan of what they're doing.
5:12
Fantastic team. That got a lot of attention. Okay, great rack for Ineffable. But actually people just loved seeing the pictures of the fiber and the sort of the fractal nature of the fiber. One of the most popular posts that we've ever had actually was around that. So yeah, absolutely. We're really excited about that. Yeah, I had goosebumps seeing the picture.
5:32
How do you measure and hold yourselves accountable in the middle of this big build out? We were talking right before the show about how flops is a vanity metric and you prefer an alternate metric. Can you say more? Yeah, so I think one thing to note is that whether it's flops or pick your other favorite chip-centric metric, these are in theory.
5:54
In other words, under some conditions, for whatever chip you have, this is the maximum amount of flops that you can deliver. What we really care about in the end is what's the performance delivered by workload. That's what we're looking at. And rarely, it happens, but actually relatively rarely, is that performance determined by a single chip.
6:14
It's flops, it's HBM, it's SRAM capacity, et cetera. Those are all super important, but it might be how two, four, eight, 16, 1,000, 10,000
6:26
more of these chips composed together with not just the accelerators, again, whether TPUs or GPUs, but then the CPUs that might feed them the data, the network that connects them all together. So what is the workload that you're running and what is the performance of that workload? One measure that is interesting is what's your Flop's utilization? For example, for a particular workload, if you have in theory a teraflop or a petaflop capable for your workload, what fraction of that are you delivering? Now, that's a measure of good put.
7:01
What is good put? You can think of throughput, which is a well-known term. That's the throughput that is possible, but now let's consider some other considerations. One is, what is the slowdown of the workload? Again, just inherent to the workload, but other aspects of it that really hits our reliability. In terms of how we hold ourselves accountable, if we have a chip failing for a synchronous workload, and many of these workloads, whether training or serving or agentic workloads, they're synchronous.
7:29
There's many, many components that are working together simultaneously. Now, if you have, let's say, 1,000, 10,000, 100,000 of these components working together simultaneously and they're really needing to coordinate at microsecond or millisecond granularity, one of them fails, it might actually bring the whole thing to a stop because everyone is counting on everyone else to do that part of the job in order to come up with the answer to a really tough question.
7:56
One of them stops. Okay, now we have to figure out what happened, which one stopped, what's the checkpoint that we have of the computation at some previous state in time. How do we restore that checkpoint? How do we restart? Worst case, we have to restart from the beginning. That would be really bad, but that can happen in certain cases, especially more on the inference side.
8:16
The point here is that if you now then have to go back and redo a bunch of computation, If you then have to pause and wait to figure out what happened with the failure, all that work is work. It's not actually helping you get the answer. In terms of, let's say that you're working out a problem on paper, step one, step two, step three, step four, if you have to go back to step one because you've made a mistake, Yes, you're still doing work.
8:42
That's throughput. You're doing work, but what is the good put in delivering your answer? It's basically the total amount of time it took you to solve the problem. That's what matters. And if you have failures, if you have failure recovery, if you have whatever it is that's interrupting your work, that's part of the issue. So now, how do we hold ourselves accountable? delivered good put, not theoretical benchmark throughput or goodness in theory, but actually for a workload for real failure conditions was happening.
9:12
And the unfortunate reality is that at 100,000 accelerator scale, I was going to say, at that scale, something's failing all the time, right? All the time. Each one of these, as you're aware, each one of these chips is a wonder of nature. They're at the very bleeding edge of what's possible to manufacture. Again, to me, it's stunning. Now, actually, these chips aren't just one chip.
9:37
They're packages made up of, in many cases, two, four, eight, maybe more chiplets that are composing together. And then, of course, HBM off to the side, network connectivity, maybe things like co-packaged optics. No criticism. Lots of things can fail. And now if you've got 100,000 of them, something is going to fail. You have to be prepared for that, detect that in near real time, recover from that in near real time.
10:02
Doing all that, it's really like finding the telemetry problem is massive. It's like finding a needle in the haystack continuously across certainly seconds and minutes, but for some of these jobs, hours and days and even weeks, just continuous and online. How we hold ourselves accountable is what's the good put that we deliver for the workloads that actually matter in the data center.
10:25
Is Google term or is it an industry term? It's a Google term, but I think more and more people across industry are starting to pick it up. And then just to calibrate me, 100,000 accelerator scale, are we talking it's going to fail once a minute, once a day? 100,000 accelerators. So let me see. I think that it is definitely going to be at the scale multiple times a day.
10:47
And perhaps depending on the exact configuration, multiple times an hour, something is going to fail. And what are the most common failure reasons? This is the problem. Actually, it's a very good question. If there were a common failure reason, we'd have figured it out and fixed it. It is a long tail of constant discovery. Whenever we have a new product that's being introduced, there's going to be something that hits us.
11:12
In many cases, frankly, because again, it's at the very bleeding edge, it might be network related. It might be how we connect these things together at super high speed. That might be something to do with the hardware. Those things we work through. But then a lot of the issues could be software. So this is the other aspect of how we hold ourselves accountable.
11:31
Again, the chip might be capable of a certain level of flops. But if you have a compiler bug, a runtime bug, a model issue, something else, operating system issue, it doesn't matter. that that's going to impact the N10 system performance. So you might have perfect hardware, fully reliable, but then you might have software issues that hurt you.
11:53
Is there a standard reference stack from the accelerator companies, from NVIDIA or from the TPU team of, you know, here is the optimal system to build around our accelerators. And as long as you build to that system, you're good? Or how much of your own kind of data center design do you have to do above and beyond what the semiconductor companies give you? Right.
12:14
So there's a reference stack. And I mean, I would say that NVIDIA is an incredible whole systems company. They're obviously a semiconductor company, but they're not just a semiconductor company. They give you a really, really strong reference stack. But I would say that many, we find that the way I'd put it is most of our customers leverage that reference stack, but many also specialize it.
12:38
So in other words, they might find that as natural for their particular use case, they have an optimization opportunity or something different that they need to do. And they're going to do that. Similarly for the TPU side, we have a reference stack. But then most people will leverage that. Many will specialize it as well. Got it. And then I guess in terms of overall capacity, you've told your teams that Google has to roughly double the serving capacity every six months or so.
13:03
Is that right? Well, so to be clear, this is in terms of effective available capacity from the way in the end that we look at it is from a serving perspective, it's token generation capability. So capacity, and this is where I would say it's a combination of software and hardware. The hardware might have some quote-unquote inherent level of flops.
13:27
What I'm saying here is not necessarily you have to double the number of flops every six months. That's one path. You have to double the capability of that hardware to generate tokens every six months. And as much or more of that is going to come from software as it is from hardware. So in other words, it could be a model optimization that delivers that.
13:48
It could be some runtime optimization that delivers that. It's really probably going to be dozens or hundreds of individual optimizations that are just landing again and again and again to make it all possible. But yes, the rate of capacity improvement is incredible. I mean, I guess we've had a few years of data center build out now and a few years of model progress and software progress.
14:10
What has been the empirical breakdown of how much capacity, maybe as measured by intelligence per watt, how much capacity increase has come from the silicon itself versus the models versus other software and any other kind of big components? Yeah, it's a really good question. I don't have the exact breakdown, but I would say that in our experience, most of the benefits come from model-side improvements in terms of intelligence per watt.
14:38
And by the way, intelligence per watt is a fantastic metric, really focusing. We also, for us, it's good put per watt, and we can come back to why the watts need to be the denominator. Good put could well be a measure of the intelligence delivered per watt. In other words, good put is a workload-specific metric. But I would say most of the gains often come from the model side.
14:59
Software system, quite a bit. Why? Because they're able to actually make sure that the hardware that you have is being used effectively. And then the hardware, it is pretty stunning. In other words, we're living in a world right now where 2x or more year-over-year performance improvements is absolutely possible. And so the hardware does support it and it is a, if you want, it's a free multiplier, not free, but everyone else above it can count on it in terms of that being a multiplier that lifts everyone else up year over year.
15:33
I'd love to talk a bit about the TPU program and co-design. So Google started building custom silicon more than a decade ago. It was a contrarian call at the time. How has the TPU program evolved Quite a lot. I mean, when the program started 2013, it was really a contrarian call. And I mean, I think that at the time, it's hard to put yourself back into the moment 2013.
15:56
Conventional wisdom, you know, all the smartest, wisest people would say, you don't build a custom built accelerator for a single workload. Why? Because Moore's Law is there, doubling of performance every whatever it is, 18 or 24 months is there. You get to leverage standard programming models, all your C++ code, et cetera, Java, Python, whatever it might be.
16:16
Like the bitter lesson of chips. Yes, exactly. It's like the bitter lesson of chips is that specialization never wins. But in this particular case, it was we have this one application. or a few, a small number of them, that would benefit tremendously and that would require an unimaginable amount of general purpose CPU to support. So really, in 2013, it was a bet.
16:39
And there were quite a few people, even within the company, I would say, who were not sure that it would work out. So it was a bet. It turned out to be a massively successful bet. So the first chip was all about inference. The second chip was, hey, actually, we can take the same idea and build a training chip.
17:01
From there, it got picked up for more and more use cases. Around the time the second chip came out, transformers were invented. This was a major, major moment, and that actually completely shifted the TPU program. We, of course, learned that recommender systems could run really, really well on the TPUs, and so ads and related use cases came along.
17:23
I would say that change and evolution has been one of expansion in scope and impact. In other words, it started with a really impactful use case of a couple of applications for inference serving. This was language translation primarily and voice recognition. to then training, to then transformers, recommender systems, and continuing to generalize as, I guess, the Gen AI moment took a hold to ever larger, ever more scalable systems as well.
17:56
You mentioned the moment the transformer came through being a major moment. I'm curious to explore, the TPU was like specific architecture, but it wasn't, you know, specific to the transformer specific. How do you kind of straddle the fine line of, you know, how specific you want your chip to be for the workload? This is a great question, and I think that it really comes down to the applicability.
18:17
I mean, I think that of the chip to a particular workload. So in other words, in the end, we've been thinking about this for every generation. Would we further specialize? The question that we were facing, let's say, two years ago or a little bit more was, in 2026, should we have two chips? or one. We could have one chip that could do inference and training simultaneously and do both quite well, or we could have two chips, one that was further specialized for inference and the second one that was further specialized for training.
18:51
This analysis and this work led to the release of two chips this year, 8i for inference, AT for training. And in the end, what we realized is that it wasn't the case necessarily a couple of years ago, but by 26, we saw inference and serving really taking off. And so having a chip that would be significantly faster for serving that we thought might be 30, 40, 50, 60% of the market in its lifetime.
19:18
started making a lot of sense for us relative to if inference if we projected inference to be two percent or five percent of the market even if that specialized chip is let's say 2x faster it might not make sense right because you'd actually go for the one general purpose chip that's not fully optimized but that that's okay because you now have the uniformity etc So really, it does come down to a calculus question of, okay, would you further specialize? How big is that workload? How big is that workload projected to be in a couple of three, four years? Is this a sustained growth for a particular workload? In hardware, again, as you know, there is this opportunity where
20:02
the more you specialize to a particular workload, the less flexible it is, the faster, the more power efficient the hardware is going to be. so it is this art and it's this projection of what are you designing to and how persistent is that workload in other words if it's going to go away after a month or two months or three months even if it's big for those three months you got a really narrow window to intercept it so it has to be somewhat durable
20:30
and you have to be able to project exactly what when you can get for specializing to it So the rough trade-off is there's a large fixed cost to support a new program. Yes. And you basically have to think that there's going to be enough demand for that specific program to justify the cost. Exactly. And to some extent, for example, for our 8i and 8t chips, both chips can do the other workload.
20:54
This is also key. If 8i could only do inference, and I couldn't do training at all. It had zero performance for training. And then vice versa, if AT was amazing at training and had zero performance for inference, that would have also been a key limitation. Why? Because we would have to predict ahead of time, over a six-year period, lifetime of the hardware, exactly how much we need of each.
21:16
In this case, it was nice because both ships are better at what they're specialized for, but if you needed to, if you had leftover capacity in one place or the other, both can actually do the other's job. Depending on the specialization you do, you might get so specialized that you actually limit yourself from being flexible, fungible. And if the rough trade-off then is kind of size of program on the other end, like it seems to me that so much of the modern AI market is transformer based.
21:44
And so maybe just asked a provocative question, why not just burn the transformer architecture into the chip? Yeah, so I think that it's transformer-based, but then the next level of question would be, transformers in the end are about the vector and matrix multiply operations. And a softmax, there's a range of essentially linear algebra primitives.
22:07
We have those, and others do as well, roughly baked into the hardware. It's all transformer-based, but then your model architecture is, okay, how many layers do you have? How do you go through the layers? How do you go across the layers? For each dimension, what exact shape of matrices and vectors are you applying? You could go further and specialize to not just the transformer, but your model.
22:30
That's the next level of specialization. I think there are a number of companies out there that are thinking about that. I think it's a very, very interesting direction as well. Okay, so at this point in time, do your customers view TPU and GPU as roughly fungible equivalents or is there a certain set of problems that is better suited for one or the other? There's for sure a better set of problems that are better suited for one or the other.
22:53
I mean, GPUs are more general purpose than TPUs for one. That is clear. They're incredible products at Google and Google Cloud. We sell a lot of GPUs. We use GPUs internally. But I think it really then comes down to the specifics of your problem. So I think that our customers, there is overlap between the two, but our customers then evaluate the workload.
23:17
and evaluate their options. What we like to do at Google is give our customers choice. In other words, we want to have the right solution for their needs and, of course, provide the solution that best meets their needs for them.
23:31
What is the case for co-design from the chip to the network to the software? And then what is the case against co-design? Yeah, so there's huge optimization opportunities with co-design. And so what you can imagine is that if you have a end layer stack and you want to be able to pick and choose whatever component you want, let's say you're running across many clouds or you're running across many pieces of hardware, many pieces of software, you could design abstraction layers.
24:01
that would basically say I can run on any hardware, I can run on any software, I can run on any network topology, any amount of network that you give me. No problem. And my system is going to be fully adaptive. Very likely, so now you have a great capability. You can move anywhere. Like new capacity becomes available, overnight you're up and running because you've designed your system that way to actually be able to take advantage of anything.
24:27
You have not hard-coded or specialized at all to anyone's particular infrastructure. Downside is you probably leave a lot of performance on the table if you become fully fungible, if you become fully flexible to anybody's hardware, software, network, storage, compute, etc. stack. So the case for co-designing is between each one of those layers,
24:50
there's a big impedance mismatch if you try to get full generality. And so there might be 10%, 20%, 2x across each of these layers. You start multiplying those optimization opportunities through, and all of a sudden you're left with a big end-to-end opportunity in terms of whatever you want to say, intelligence per watt or good put per watt.
25:12
that you can leverage, even again, all the way down to power delivery and power availability, software optimizations, et cetera, et cetera. So pros are you can run anywhere, anytime. You have no lock-in.
25:27
Cons are you're leaving significant, not likelihood, significant amounts of performance on the table. So my understanding is that if you take OpenAI and Anthropic, OpenAI was kind of primarily building on a homogenous compute stack and Anthropic was building a roughly more heterogeneous compute stack. Do you think that's part of the reason, that code design is part of the reason why they converged on what is rumored to be very different architectures for their models?
25:56
I can't speculate. I don't want to speculate in terms of what OpenAI and Anthropic are doing. It's just one possibility. But I think that
26:04
without knowing the details of what they're doing, I would imagine that there could be many reasons for why they wound up with different architectures. Well, what about for Google then? I'm curious what the working relationship looks like between your team and DeepMind, kind of who's in the room, at what stage of model development, and how are you making decisions together on co-design? It's one of the most fun and frankly gratifying parts of being at Google is the opportunity to work really shoulder to shoulder with the DeepMind team.
26:33
in terms of co-design of our hardware and models there's a third element to it in terms of that we also get to extend that with the consumer services and cloud i'll put that aside for a moment i can come back to that but with respect to deep mind it it really is a deep partnership so let me i mean from the past i can give you examples where
26:53
they have come up with model optimizations
26:58
let's say to transformers or to particular math that they might want to do. And we might have a chip in progress, not quite done. But then they say, oh my gosh, if we had hardware support for this,
27:12
our end-to-end training or serving might get significantly faster, more efficient. What would it take to actually now change our hardware definition that we might be in execution on to accommodate? And this then leads to our engineers and researchers getting together intensely in the same room for a few days, a week, two weeks, saying, okay, yes, we can do this.
27:39
More likely, we couldn't quite do what you wanted. but we can do this other thing. And then maybe you can change your model architecture in this other direction that gives you 98%, that gives us 90% of what we're looking for. And yes, we can then go back and intercept the hardware. And, or we can say, you know what, we're going to delay the tape out by a week or two weeks because wow, for this level of benefit, it's, it's totally worth it.
28:05
You know, similarly when we're, um, projecting our roadmap out and it's time for, you know, we have many generations of chips that are essentially in progress at any point in time. We have the chips that are in production. That's one. We have the chips that we're working on getting into production. They're already back from the manufacturer and we're debugging them, making them work.
28:25
We have the chips that are in implementation that are about to tape out and go to the manufacturer. We have the chips that are in design phase, and then we have the chips that are in concept phase. So it's really this five, six stage, many, many year pipeline from in production to in your mind, et cetera. The collaboration with DeepMind in production is significant because we get to work together in maximizing delivered intelligence or delivered good put per watt.
28:55
We know exactly what's happening in the model and we know exactly what's happening in the hardware and everything in between. We also get to collaborate deeply on though the chips that are actually just about to tape out. Why? Because we can intercept and we can make changes to the chip, literally the chip architecture in flight, which would be somewhere between hard and impossible to do if we were working across company boundaries.
29:19
Not impossible, but it would be much harder for us to say, oh my gosh, we're a few weeks or a few months away from getting this chip done, and now let's get in the same room, shoulder to shoulder, and figure out if we should disrupt the program. It's possible, but harder is what I would say. And then, of course, for the chips that are in design, we have many architectures we can evaluate together.
29:40
We can then ask our colleagues in DeepMind, where do you see model architectures going? in two, three years time. Here's the Pareto of things that we can do. Here's the Pareto of where model architecture is going. We have actually deep and significant simulation infrastructures that can predict how the workloads are gonna map to different hardware architectures.
30:03
deep, deep and fast iteration. It really isn't the teams are working separately. It's in the same building, same rooms many times, deep daily interaction. I talk to whether it's Cori or Demis multiple times a week, et cetera. So it really is a super fun aspect of the work. That's awesome. And eventually their models will help with chip design.
30:27
We're using Gemini to design hardware for future Geminis as well. That's really cool. I want to come back to that a little bit later on the collaboration itself. I guess, is there an impedance mismatch still of just, I figure the cycle times are probably just slower when you're dealing with hardware than your DeepMind folks get to deal with on the software side.
30:47
And I figure your planning cycles are much further ahead. For sure. And so how much room do you actually have to adjust Yeah, it's a really good question. We are planning hardware two, three, four, five years in advance. No question. In other words, we talked about TPU8, I and AT just now that we announced, but you can imagine that 9, 10, and maybe some others are in concept execution to something else.
31:12
And they might be years out. If you're working with model architecture, by default, you're not going to be thinking years out. But I think this is also the great thing about how the company has grown up together. And in other words, Google Research and DeepMind and Transformers being invented, all of this was happening in these overlapping rooms as well.
31:32
So in other words, there's a whole generation of researchers who've grown accustomed to being able to influence the hardware. and knowing that the hardware is operating on multi-year cycles, and they also know that, look, if they have a small tweak that's going to deliver like 1% or 0.5% or something like that for a chip that's about to tape out, they're probably not going to come to us because they know enough to know that actually it's not like software where you can just do a change list and it's going to ship to production in two weeks' time.
32:01
Like actually stopping a tape out is a big deal. But they also know if they've got a really good opportunity,
32:11
yes, we're absolutely going to work together to figure out if we can get it in there. So as I said, it's multiplicative in terms of where the benefits come from. The hardware does lift all the tides, right? And so significant portions of the DeepMind team, we're so grateful for it. It's an amazing team. But significant portions of it are thinking about How do I influence the roadmap? Because it's actually a pipeline.
32:34
Like the idea I had two years ago, like it's in production now. And it's helping all the workloads at Google go faster. Like that's a good feeling. Totally. It seems like an impossible task, though, to predict in five years what workloads will be most common and what algorithmic breakthroughs will have occurred.
32:53
Doesn't it seem like an impossible task? I hear you, that it would seem like an impossible task, but here's the awesome thing. And actually, we're working on writing this in great detail, and it's a lot of fun to do it. The stunning thing is that the TPU architecture, at a medium level of detail, not at a super high level of detail, at a medium level of detail, hasn't really changed since TPU v1.
33:16
One way to look at it is the instruction set architecture for a CPU. You've got loads, and you've got stores, and you've got ads, and you've got subtracts and branches, and yet what has the software on top of it done over that period of time? Same thing with TPUs. We have some fundamental instructions and fundamental primitives. Of course, we've extended it.
33:37
It's not like the instruction set hasn't gotten changed at all. But the primitives there in terms of specializing the numerics to very big, very large matrix multiply units, a sparse core that manages vector operations and scatter-gather operations, etc. There's five or six things. that really define a remote load store. That's another one, actually.
34:01
We can read and write remote memory through our ICI network. The fundamentals have been there and they've extended to many generations of models, many generations of even deep neural network algorithms and model structures. Okay, speaking of shifting workloads, it seems like one of the biggest changes in workloads over the last year or so, I think it really started at the beginning of this year, this calendar year, was kind of the rise of the long horizon agent.
34:27
Yes. And I would guess that's a very different shape of workload than kind of the quick turn LLM conversations of years past. What does that mean in terms of data center needs? Yeah, so I think there's two huge aspects to this. One is it's no longer human to, whatever you want to say, accelerate interaction. In other words, when you are typing at a prompt on a web browser or on your phone or whatever, of course, in response to your prompt, there's going to be a bunch of work that happens.
34:58
But then when the response comes back, you've got to read it. You've got to think about it. And then maybe you have a follow-up. That's going to be multiple seconds of interaction time. Now on Deep Horizon Agents, It's not, there's no human loop that is going to naturally rate limits how quickly requests are going to go to the model, right? So this, what went from, you know, seconds, maybe tens of seconds in terms of interaction time is now going into perhaps milliseconds, right? As soon as I get a response back, I can parse it.
35:27
I can reason about it perhaps a bit, and then I can figure out what my next request is going to be. So that's big change one. Big change two is all that reasoning and all that parsing is going to probably happen on a CPU. And that CPU is then going to have to probably think about, okay, what other state do I need to go gather before making my next prompt back into the model? Like in other words, I've learned something from this response.
35:50
I'm going to take another step, but I actually need to go grab some context, maybe from DRAM local to me, maybe from someone else's DRAM on another CPU, maybe on SSD or maybe on HDD somewhere else. So now a huge amount of orchestration has to take place as well. So, the design actually is changing pretty significantly where the demand for accelerated compute is going up, but the demand for CPU and networking and storage, the traditional data center, CPU, et cetera, is also going through the roof at the same time.
36:25
So, does that mean you're putting more CPUs alongside your GPU racks then? Yeah, GPU, TPU racks. This is the key question is going back to this question of optimization and specialization. If we start putting lots of CPU racks, and this is a question, next to the GPUs and TPUs, that means that actually we can't fully specialize to the density and network requirements, let's say, of a TPU rack relative to a CPU rack.
36:53
A TPU rack is going to be more dense than a CPU rack. It's probably going to need more networking than a CPU rack. So in other words, now our building design is going to change. Another option is maintain your uniformity. Put your TPUs or your GPUs all in one building. and the building next door, maybe, put your CPUs. And maybe, by the way, the hard drives have to be in another building on the other side because they have yet another set of requirements.
37:21
Now you need networking between these buildings at a pretty significant level. So, once you leave a building, the networking complexity goes up significantly. From a reliability perspective, from a cost perspective, latency goes up probably acceptably, but still now you might go to hundreds of microseconds, potentially more with queuing between the components.
37:45
So, the considerations do change in pretty interesting ways. Your job is hard. That's fun. It's fun. Yeah. What's happening on the networking side? And I've heard that Google's always been at the forefront of the newest in networking, including optical. Can you say a word on what is the state of optical networking? You know, we at Google, this was probably 15, 16 years ago, were among the first to bring essentially what's called wave division multiplexing, where you could put multiple signals on a single fiber within the data center.
38:21
And we actually leveraged that for all of our communication between racks. At the same time, we introduced technology along with wave division multiplexing called optical circuit switching. And essentially what optical circuit switching does is in contrast with traditional electrical packet switching, what it does is it transmits and moves data entirely in the optical domain.
38:45
The great thing about that is, and so let me describe the technology that underpins it. In a packet switch, You would take a packet that has a header. You would look at it in the electrical domain, figure out where the packet is headed. In this header, it might have an IP address that says, okay, where do I head it? You look up, okay, for that destination, you look up in a table, what port do I forward it along? So basically, billions, billions and billions, perhaps trillions of packets coming through at super high speed per second, you're forwarding them along.
39:18
Optical circuit switching says, I'm not touching these bits in the electrical domain. What I'm going to do is I'm going to figure out for an input port, which output port to send the light to. Okay, and now I have, there's multiple ways to do this. The one that we started with is called MEMS switches, Micro Electrical Motors, that basically control mirrors in 3D.
39:42
So we can now programmatically take a box that might have, you pick, 128 ports, 256 ports, some number like that, And we can configure every input port for a fiber that comes into it, map it to an output port, and then change the rotation of mirrors where the light literally shines down on these mirrors and gets reflected to the right output port.
40:02
Initially, the reason we did this twofold was to create locality between groups of racks. Let's say I had a compute cluster and a storage cluster, and both of them were in support of, again, making it upsearch. We knew that these two clusters would talk to each other a lot, so we would configure the mirrors to create shortcuts purely optically between those two clusters, clusters of racks.
40:30
That was reason number one. Reason number two was we wanted to be able to expand the network and contract the network without actually moving any fiber. And without working into the details, I could draw this on a board, the optical circuit switch would allow you to actually reconfigure the spine of the network. to expand it or shrink it without a human being having to do anything other than be literally a controller that would manage that.
40:53
Now, fast forward to TPUs. A TPU has a torus topology that connects all the TPUs to one another directly.
41:04
I talked about throughput and good put earlier. One of the things that we can do is if we have a TPU rack that fails, we can replace it with another TPU rack without moving any fiber. Again, I'd have to wave my hands or draw the whiteboard. But essentially, we can say we have a spare rack available at all times. And when a rack fails, we redirect the light to that new rack.
41:29
And that can be done in milliseconds. Why do you have the fiber at all then? As opposed to fully free space. Yeah, it's a very good question. Attenuation loss and the bandwidth would drop dramatically. Furthermore, across the range of a very large building, aiming everything without the benefit of fiber to connect it all together in 3D would be challenging to possibly impossible.
41:56
We've talked about it. We've talked about it actually, and there have been some really interesting discussions. But yes, it's mostly in fiber, but then when they hit the optical circuit switch, essentially the light then literally shines down on these tiny chips. So that's one big direction of networking, but there's a lot. I mean, I would say networking is exploding in terms of its capabilities and need, frankly, in the data center.
42:22
So cool. We could have an entire conversation on that. Yes, it is very, very cool. Even like going back to the picture from Ineffable, it was all the cables that cut over these eyes. Exactly. Yeah. And those cables, eventually some of them wind up at optical circuit switch in our data centers. Makes sense. Okay. I want to flip to talking about power.
42:38
You've been talking about good put per watt and other units per watt. Makes me think per watt means power is kind of the binding constraints or the scarce constraints or the expensive constraints in some way. So I get asked this question, what is the biggest constraint that we face? And the reality is there is no single biggest constraint that we face.
42:55
They're all constraints. They're all super hard. And they shift continuously. But if I had to answer fundamentally, I would say that power is the single most fundamental constraint that we face. Like everything else seems...
43:11
We know how to solve them, and it's a question of solving them over some period of time. Power, yeah, I think the way that you put it is really nice. It's a binding long-term issue that we do not have a, I mean, nuclear, perhaps nuclear is going to, abundant clean energy would solve a lot of problems for sure. When that happens and when it happens at scale, still unknown.
43:38
And so how does it work in practice? You know, you're standing up a new data center. It needs a gigawatt of power. I imagine you can't just call PG&E and say, Hey, please send a gigawatt of power. So what does provisioning power actually look like? And are you having to actually radically integrate all the way down to like, you know, doing your own turbines or how do you solve the power bottleneck? yeah it's again it's a it's a big question uh important question we do uh our our preferred model at google is always to be uh utility connected to be good connected so it's the equivalent of calling your um uh so you can't just call pg e exactly you call uh very politely your
44:17
favorite utility wherever it is that your data center is and of course you give them many years of notice. So in other words, if we're talking about a gigawatt scale, it's not something that you can say, hey, I need a gigawatt tomorrow. When can you start billing me? It's something that we co-plan together. For us, it's something that we also take very seriously from the perspective of ensuring that when we work with these utilities, the costs of putting that infrastructure in place we could get into a long conversation just on this topic as well, that we cover those costs because with the way that billing works, it actually could be that by the act of the utility
44:52
building up capacity for us, let's say, other people's rates could go up. In theory, what we ensure is that actually, let's say the transmission lines that have to be built, upgraded, additional utility base stations, et cetera, that we pay for those as well. But so it is a long planning process. It can absolutely be the case that, let's say that we need a gigawatt in, I'll make up a date, 2028, and the utility can get us a gigawatt in 2029.
45:21
They can get us, let's say, 700 megawatts in 2028. So now we might be left with a question of how do we cover those 300 megawatts? Well, one answer is just wait. Another answer is to say, okay, well, would we figure out how we generate some subset of that power ourselves? And maybe that would be with solar cells or with batteries as backup, et cetera.
45:43
Would it be for other sources? So then we, again, work with the utility. So it might also be a combination where we might maintain some power generation local. and have the capability even when the utility comes online fully, let's say the gigawatt scale, where we could actually provide power back to the grid. So having that when they need it as well.
46:05
So in other words, there might be the one way to look at it is there might be the two weeks of the year. Maybe it's the hottest two weeks of the year where there's a huge amount of residential demand. If we have some local generation, we can then provide that back to the grid as well. So it really is working over multiple years with the utilities for us.
46:24
Why is your preference to do that versus to kind of go like to work with the utilities as opposed to, to kind of vertically integrate yourselves? Yeah, so the main reason is flexibility and uplift on both sides. So one way to look at it is statistical multiplexing, or if you want the law of large numbers. If we need to have a gigawatt of power, let's say we wanted to have that with 99.99% plus reliability.
46:53
Probably means we have to build two gigawatts of power. At that level, 99.99 or 99.999, you have to have one plus one redundancy.
47:03
that gets expensive. And then making sure that that's going to be ideally a clean energy source, probably right next to our data center, that can also be challenging. Now, if we partner with the data center, again, maybe we have some amount of that that we can bring ourselves, that we can give to the grid when they need it less, that we can take their power.
47:22
So in other words, by leveraging that statistical multiplexing of our much larger base, actually everybody wins. We win, the grid wins, residences win, etc. We prefer that. And in rare, rare cases, we will do it, let's say, behind the meter. But even then, we're doing it under the assumption that we're going to work with the utility where it might be a year later.
47:45
We want to be connected to the grid. Again, it's an uplift for us, it's an uplift for the grid. Makes sense. How did you decide how big to make a data center? Yeah, that's an art, and it's a source of significant debates. At some point, we had a debate, I remember, even 10 plus years ago, 15 years ago, big debate at Google, should we just put everything into one data center? Back then, it was going to be a gigawatt.
48:08
Simpler times. Yeah, different times, like 10, 15 years. We're going to have a gigawatt data center. So one obvious concern there is single point of failure. If you're looking at things from a 30-year perspective, that's a huge concern. On the other hand, if you're talking about training workloads today, bigger is better. In other words, because from a networking perspective, actually, you want to have things concentrated in as small a distance as possible.
48:35
But then again, two issues, single point of failure, but then also now power availability. Well, a gigawatt was huge 10 or 15 years ago, but imaginable. There's no way anywhere in the country or the world we're going to get the total requirements of Google built in one place. It's just not going to be possible. So now what's the optimum size? Again, we have models for this, simulators, et cetera.
49:00
But it also depends on, look, in some places we're going to be at the edge of the network or we're going to be in country. That might be tens of megawatts. we actually partner with ISPs, that might be a rack. Like literally, it could be a rack. This is a training cluster. Okay, maybe that's going to be closer to a gigawatt. Some other sites might be hundreds of megawatts, et cetera.
49:23
Interesting. I'd love to understand how you think about the portfolio lifecycle management. So I guess my guess would be that the biggest, newest clusters are used for training the latest frontier model. And then you kind of recycle the other stuff and run inference on it. Is that like a fair framework? Are you also standing up inference specific clusters? How does that all work? Very reasonable framework that you're thinking about, and I think makes a lot of sense.
49:51
But I would say that the demand for inference is such that we can't just rely on whatever older training clusters that are no longer being fully used just for training as the basis for inference. And also, if you think about it, we might centralize again in a particular year into a small number of large sites. training clusters and there is benefits to keeping the network distance between them small so now that's wherever they're located in the world is they're probably going to be let's say on the same continent or they might be even the same portion of a continent in a particular year so now you might be left with not enough serving capacity on the other continents etc
50:31
so then okay actually we didn't have to go build the specialized inference elsewhere across the world. So I think your intuition is spot on, but not wholly. In other words, it really does have to be, okay, here are the training clusters. Yes, probably they're going to be used for serving in some number of years, but then we're also having to build the serving clusters as well.
50:51
And your serving clusters, are they different than the training clusters? Are they smaller? Are they cheaper per megawatt? Not necessarily cheaper, actually, because for serving, there is more need to co-locate compute networking and storage. Then we get to the inability to specialize. For training, you actually have this big density, uniform deployment, etc.
51:16
Whereas for serving, you're going to have to have the mix of storage, compute, and accelerators. Furthermore, it's a very interesting aspect of this that we could go into more detail on as well. For serving, you actually don't want to have too much in one place. You want to be serving your workloads from all over the planet. But now we wind up having individual model endpoints, and there actually can be variations of models.
51:43
So now we have to spread these models out across the world, also accounting for locality. So they will be smaller. They'll actually be less vertically integrated, probably, on the inference side. Super interesting. So if you made a data center five years ago, state-of-the-art accelerator five years ago was very different, vastly less efficient than the ones that are being made today.
52:06
Yes. Do you actually go back and swap out the chips in those old data centers? And I know this relates to there's been an ongoing debate, I think, of what is the actual useful life of a chip. Yeah, so I've been on record as saying this in one of my more, I was surprised by how much pickup this had, but I said that our seven and eight year old TPUs are still at 100% utilization.
52:30
Wow. I think I saw this, I didn't realize this was from you. Yeah, I didn't mean for it to be a meaningful statement, but it was a meaningful statement, apparently. Yeah, so our older TPUs, GPUs, but our older TPUs are seeing significant utilization We do replace them. In the end, there's a question of, OK, the depreciation lifetime is approximately six years.
52:53
So once they're fully depreciated and given the power efficiency of newer generations, et cetera, it does make sense to replace them and upgrade them. It's not that we do replace the chips, but it's really the systems. So in other words, we think in terms of pods. So a TPU, let's say 8T pod might be 9,600 chips. And it might be 140-something, 152 racks, et cetera.
53:17
So then we would say, OK, we're going to pull out that pod. And we're going to replace it with whatever it is, TPU 12 or 13 or 14 pod. It might not be a perfect fit. Then we have to, in other words, the new pod's footprint might not be a perfect fit for the old pod's vacancy. We then just have to account for that. We have to figure out how we would retrofit.
53:39
We can't plan that because we don't know what that many generations of TPUs are going to be. So then it's, again, a bit of an art to figure out how we would, and it's actually hard work, figure out how we would decom and then replace as quickly as possible with new TPUs. You've written publicly about open standards. Can you say a word on that? Yeah, so I think that we talked about this interoperability question.
54:03
And so for us, while we support vertical integration and we make it possible to extract just as much performance as you want, for us it's really important to not force lock-in and not force a sort of walled garden in terms of how the system works end-to-end.
54:23
I'll give one example. At Google, we've developed a model development framework called JAX. We like it a lot. We think it's really, really good, and we use it extensively internally. Many of our customers like PyTorch. And so one thing we could say is, hey, if you want to run on TPUs, you got to use JAX. It's the best. I'm being facetious a little bit, but maybe it is, maybe it isn't.
54:49
We think it's the best. And that's your only choice because it's so great. Or we could say, look, if you like JAX, we like JAX. If you like JAX, you can use it. But if you like PyTorch, we have Torch TPU, where your unmodified models, et cetera, can run.
55:07
We've seen this throughout history, actually, with many, many examples. I've used the example of IP. Why did the internet protocol win? There were actually many competing protocols to IP. This is ancient history in the 70s and early 80s. Why did IP win? It was because it was open standard and it was this narrow waste to the hourglass where anything could run on top in terms of software and anything could run underneath it in terms of hardware.
55:35
So open standard, interoperable, anyone who brought IP could plug into a router port and become part of the internet. It was beautiful. And that's what allowed the internet to explode in its growth and reach across the world. So we really do believe in those open standards in terms of plug-in points. Now, if you want to plug in something highly specialized...
55:57
You can. If you think that you've got a better thing to plug into our framework, you absolutely can. But we want to really support open standards, ideally open source around it as well. It's too big and too important of a build out to be any one vendor's closed proprietary stack. Yeah, we really believe that. And it's got to be one of choice.
56:17
And that's also why, for example, we fully support and have TPUs, GPUs, other accelerators, et cetera. Yeah. How has day-to-day life for your team changed with AI? Where is it changing your function the most?
56:32
So I think, whatever you want to say, the easiest answer is on the software engineering side, where it's been documented externally quite a bit. So I think that we are, at Google and my team, using it to great effect in terms of our software development capabilities, but also, frankly, tests, rollouts, even helping with design, et cetera.
56:58
But there have been on the hardware side, which has maybe gotten less
57:03
external coverage, also significant change. In other words, my hardware engineers, I was just looking at the numbers earlier today, in fact, are using AI, using whatever the token counts, if you want, not the best metric, but it's still a metric. Our hardware engineers are using AI as much as the software engineers are. And productivity has gone up.
57:24
Time from design kickoff to tape out is shrinking. Time for bring up is shrinking. So in other words, productivity is going up significantly. But then other maybe less expected changes, how we do data center design. has changed significantly. In other words, how we evaluate, you mentioned, hey, do we do a gigawatt building or 100 megawatts campus or 200 megawatt campus? In the past, this would be very detailed, very spreadsheet-driven, human-driven processes.
58:00
And it still is to some extent, but there's a lot of AI now involved that really streamlines the process in terms of planning and development as well. Interesting. There's like reasoning models? I wouldn't say quite yet reasoning models. It's not replacing human judgment, but it's making it much easier to bring the necessary information together in one place and basically put the right information in front of the humans making the decisions.
58:30
Okay, I'm going to bring us home with two kind of out there, kind of fun questions. Question number one, orbital data centers. I've seen that Google's thinking about this quite seriously, and you don't need to comment on that. But I mean, does the fact that people are seriously running the numbers on orbital computes mean that there are kind of serious binding constraints on Earth? And what do you think of orbital data centers? Yeah, it's an exciting direction.
58:55
We are actually pursuing it and we've referred to it more with no tongue in cheek as a moonshot, as one of our big, big efforts that we're excited about investing in. It goes back to the earlier part of the conversation where you raised, I think correctly, that in terms of fundamental constraints, energy and energy production is a key, key challenge.
59:15
And so really bottom line is from a fundamentals perspective, In space, you have something like 40% more power available because of lack of attenuation in the atmosphere, et cetera. In other words, it's just more solar capacity, 1.4x. So that's one part of it. But in a sun-synchronous orbit, you have 90% to 100% coverage of sunlight on your solar cells relative to 28%, 30%, maybe 35%.
59:49
on land. So in other words, just this enormous amount of power, obviously the sun, you take the 1.4x and you take the 3 to 4x in terms of number of hours per day, you remove batteries more or less from the equation, And now you have the potential for something that can really, and of course, carbon-free. Lots of benefits, lots of challenges.
1:00:14
Yep. Lots and lots of challenges. So in other words, OK, now what about cooling? Naively, you might think cooling in space is easier. It's actually harder. Reliability, we talked about how these things fail sometimes. Repairs becomes harder, not impossible, but it becomes harder in space. Networking. Robot up with the cluster. Exactly. Perhaps this model is a promising direction.
1:00:39
Redundancy will probably be your friend. Your prescience question on free space optics is now going to become reality. We're probably not going to be stringing fiber between these components, so it will actually be free space. And we're going to have the lasers pointing at the receivers and calibrating in real time. There are no showstoppers here, no fundamental showstoppers.
1:01:00
Last question. I've had this image of the ineffable big supercomputer that you built them
1:01:08
in my head this whole time. And so here's the question. 10 years from now, what will the most frontier supercomputer look like? Oh my gosh, yeah. So 10 years is right at that edge where it's, I would say in all sincerity, probably some people are thinking about what the 2036 computer looks like. We might have a few thinking that far ahead, but wow, the cone of uncertainty there is way, way too wide.
1:01:40
Like in other words, if you look at the trends, The level of integration is going to be tremendous. So we saw the beautiful fiber and the VR200 racks that we put together for ineffable. My guess is it's going to be more integrated, less fiber to, I won't say no fiber in 2036, but I think that probably from a rack perspective, the rack will look much more tightly integrated and the amount of fiber, there'll be maybe a small fiber bundle coming out of the rack.
1:02:10
And then, so in other words, one view you might have from a modular manufacturing perspective is these racks likely in 2036 are going to be manufactured centrally. And so whether they have 72 or 144 or 288 or probably more, 576, 1152, pick your favorite multiple of GPUs, since we're talking about the ineffable case, but then be the same for the TPUs, deeply integrated into a rack.
1:02:38
Would a rack be multiple megawatts? We don't know exactly how, but 2036, we get to imagine things. Could be multiple megawatts in a single rack. and now you bring water, and you bring power, and you bring fiber, right? You wheel that rack in, you plug those three things in, and you're off to the races as well. Okay, so it's not going to be like a big alien orb then? Up in space.
1:03:03
By 2036, I wouldn't say that it's not possible for 2036 for it to be then directly launched into space and then maybe
1:03:13
taken by a robotic arm at some space station and then plugged into the right module. Maybe that for 2036. Fun. Fun to think about. I really enjoyed this conversation. Yes, it was a ton of fun. This is the most intense, biggest CapEx build out in history, but I also think it's a technology revolution that is a thing of beauty.
1:03:37
You have such a grasp over the beauty of the technology and the binding constraints and how to balance this all. Google is in good hands. Thank you for taking the time to share what you're doing. It's a ton of fun. We get to live through this and we get to define it. So thank you very much. This was a great conversation. Thank you. Thanks.