The Infrastructure of Intelligence: NVIDIA and OpenAI on a Decade of Building Frontier AI

The Infrastructure of Intelligence: NVIDIA and OpenAI on a Decade of Building Frontier AI

NVIDIA

0:01 Ian, you've been up here so much today.

0:03 They should have just put you a chair over in the corner.

0:05 Nice and quiet.

0:06 Nice comfy chair over there.

0:08 So one of the things we wanted to talk about

0:10 today is the concept of building AI infrastructure at scale.

0:14 But what scale means in 26 is different than what

0:18 it meant two years ago, or even a year ago.

0:21 What does it take to build compute at scale today?

0:23 And do you want to start us off?

0:26 Yeah.

0:27 I mean, I think if you today no AI model except for maybe

0:31 the smallest ones fit on a single GPU or even a single server,

0:35 we've now gone from, uh, GPU scale,

0:37 NoSQL to rack scale to row scale to really being a data center infrastructure.

0:43 You have to design a compute token factory from end to end.

0:48 And there are huge efficiencies when you

0:51 do that in terms of coordinating the power,

0:54 scheduling the KB cache, operating and serving literally.

0:58 Um, well, Sachin is going to probably can speak more to the to the thousands

1:04 or tens of thousands or hundreds of thousands

1:06 of queries coming in to all the compute infrastructure.

1:08 It's not just a GPU problem,

1:10 it's the CPU problems with storage problems and networking problems.

1:14 So I think that has been the biggest pivot in the last, say, five years,

1:20 but certainly in the last two,

1:22 as we gone upscale to sort of gigawatt scale AI factors and which is exciting.

1:28 I think from a business standpoint, we have to help the world,

1:33 you know, build a supply chain that can build,

1:37 all this infrastructure and also understand how they can

1:40 run it efficiently to maximize cost or minimize cost,

1:44 maximize the throughput, and get the most out of the infrastructure.

1:47 And stuff's changing so fast.

1:49 It's a it is actually not a, you know, one time strategy.

1:54 It's a continuous sort of Silicon Valley technology innovation, collaboration.

1:58 And I mean, to a large extent, what's being built at this point.

2:03 The scale is unprecedented.

2:04 So are you just making it up as you go along?

2:07 How do you how do you de-risk doing something

2:10 that no one's really done before in this way?

2:13 I mean, I think it's, Really partnerships like the one we have with NVIDIA.

2:18 So, like Ian said, I think the scale continues to grow.

2:23 On the training side, scaling laws have not stopped.

2:26 The scaling laws continue to hold.

2:29 And so what that means is we will still see those jumps every year.

2:34 Every generation on the amount of compute will be

2:37 doing growing at training these next generations of models.

2:40 So as that scale increases, it becomes a whole system optimization problem.

2:44 It's not just chips, uh, GPUs or CPUs, networking memory.

2:48 It's also the power distribution system, the cooling system,

2:52 the whole warm shell with the chips inside.

2:55 It needs to be treated as one unit and that's where it's super important to.

2:59 Work with the whole ecosystem.

3:00 We can't do it alone.

3:01 So we rely on partners like NVIDIA to actually build build that.

3:05 I think as we look at this, I think very consistently we see returns to compute,

3:11 like we can see the leaps in intelligence.

3:15 We are seeing that with Astra coming out.

3:18 It is the first one trained on Abilene on black GPUs at scale.

3:23 that is obviously showing phenomenal results.

3:25 And I don't think that's stopping anytime soon.

3:28 So one example of building compute

3:30 for the future is the massive Ohio Data Center.

3:34 That is a project between OpenAI and NVIDIA and SB energy.

3:37 Um, can you tell me, apart from the fact that it's huge,

3:41 what's what's novel about it, besides the scale.

3:44 When you start on a project that's at literally gigawatt scale,

3:48 you can start from a blank sheet of paper and see where we can optimize.

3:53 So obviously we want to be able to leverage the ecosystem

3:57 and the capabilities we have today and take it to the next level.

4:01 But when you get a chance

4:03 to actually design things like the power infrastructure,

4:05 uh, as I talked about a little bit this morning,

4:08 you know, the there are a wide variety of workloads and each

4:12 have a different level of individual scale and also power within the rack.

4:17 You know, if you're maximizing user token rate,

4:19 you may be more communication and latency

4:22 sensitive than in just driving up flops,

4:25 where if you're doing offline or batch processing or large pre-filled jobs,

4:30 uh, it's more, uh, flop rich and powerful, power hungry.

4:34 That dynamic in this inside of the data

4:37 center actually is an opportunity to optimize.

4:39 You can schedule work on different nodes,

4:42 on different rows inside the data center

4:44 to distribute the compute and the power consumption.

4:47 Um, but you got to do that in a way that it can be,

4:51 uh, you can take advantage of it.

4:53 So if you're a job scheduler,

4:55 knows what the where the jobs are, how much their power they're consuming.

4:59 You can then build a power infrastructure

5:01 that can dynamically load balance the delivery.

5:04 By doing that, you know you're going to only you're going to consume

5:08 your full allocated power instead of just

5:10 sort of provisioning for the whole thing,

5:12 and maybe only use 60 or 70% of it most of the time.

5:16 Um, to do that, you have to have a smart GPU,

5:20 which can be understand its own power and be able to manage it.

5:23 You need to have a rack that actually is able to handle

5:27 transients and EDP spikes and other things to maintain to keep the, uh,

5:32 the, the electrical characteristics consistent and modeled and safe.

5:36 And then also smart, uh,

5:38 intelligent power controllers down the road, down the racks.

5:41 Uh, and all the PDAs that goes back to some,

5:45 uh, math schedulers be able to keep all that operating.

5:48 If you do that, um, you can get upwards

5:51 of 30 or 40% more GPUs in the data center,

5:53 but you got to design that from the beginning.

5:56 So that's one of the exciting things about Portsmouth.

5:59 And designing these next generation data centers is that you can think about

6:04 the problem end to end and design

6:07 in collaboration with with OpenAI in their workflows,

6:10 but also put the right things into the compute trays,

6:14 the PDU specs, even the interconnects

6:17 and the protocols for talking to the the end

6:21 of row and even up to the transformer to manage reliability and manage power.

6:26 Um, I've become a little bit of a power engineer

6:29 in my next click of my career and learning about what does it

6:32 take to actually go from being voltage to low voltage

6:36 to and kV and dealing understanding what EDP and power spikes and transients.

6:41 Uh, capacitors are a good thing.

6:43 They can help normalize power and actually keep

6:46 the whole thing flowing and the compute, you know,

6:50 on an instant without creating a havoc for a lot of things, like a power grid.

6:54 And when you when you look at all of these innovations

6:57 and things that are different in that data center project,

6:59 what do you how do you think it changes the way

7:02 the whole industry will build data centers in the future?

7:04 What do you hope people might be able to take

7:07 away from what what you're doing with your partners?

7:09 Yeah, I mean.

7:10 When it's fully built out, it will be the largest data center ever built.

7:14 Obviously.

7:15 Who knows?

7:15 If no one passes it yet.

7:18 But I think.

7:19 There's so many things there that at eight gigawatts fully built out,

7:23 it's a big chunk of the grid in Ohio itself.

7:26 And so as we think about the future, how do we co-design the workload,

7:31 the training job, for example, how do we design?

7:33 What do we need on the compute side,

7:36 what kind of reliability we need for our training job?

7:39 What kind of networking reliability do we need?

7:41 And then how does that flow through

7:43 to the cooling system design and the power system design.

7:46 All of that is one continuous loop.

7:48 The second thing I'd say is,

7:50 if designed right, this can help stabilize the grid.

7:53 In Ohio, whenever there's spikes, we can go down on our jobs.

7:58 Whenever there's gaps,

7:59 we can actually go up so that we can keep the grid smooth.

8:03 And that kind of stuff can actually help reduce electricity bills.

8:07 That sounds somewhat counterintuitive, given all the concerns about the impact,

8:10 the negative impacts on the grids right now.

8:13 Are you saying that you're doing things

8:15 that are different or that people should?

8:17 I think that is.

8:18 An opportunity to do things that are different

8:20 when we co-design like this with the whole ecosystem.

8:23 Right.

8:23 So today everything's designed in silos.

8:25 The benefit of this greenfield project is from the ground up.

8:29 We can design all of the different layers appropriately,

8:32 such that this can be a very good citizen

8:35 on the grid and actually help the grid modernize,

8:39 stabilize and reduce electricity bills by doing

8:42 adaptive load sharing with the whole grid.

8:45 So such and a lot of people spend their weekends relaxing.

8:50 But it sounds like your team spent the weekend trying out Astra on a Vera Rubin.

8:55 Can you tell us a little bit more about that and how it's going so far?

8:59 Yeah, this was uh, funnily so Ian and I had a chat last week,

9:04 and I was just telling him about how we are about to launch

9:08 a service which is going to use our AI to help infrastructure designers design,

9:14 optimize and operate that infrastructure, including, for example,

9:18 inference of any model on any chip.

9:21 And and Leon's challenge to me was, oh, great, show it to me on Rubin, right.

9:27 We have a precious Rubin rack in our lab.

9:30 Uh, it's, uh, but it's, uh, we'd be super excited to ramp,

9:34 uh, Rubin as quickly as we can.

9:36 And so we took that challenge.

9:37 So the team, uh, took Astra, uh, actually a checkpoint of Astra,

9:42 uh, and first took a Blackwell implementation of Astra inference.

9:46 Astra running on black will check and just

9:49 run it on Rubin with very minor modification.

9:52 It's a testament to NVIDIA's hardware that out

9:55 of the box we got a 3x improvement in throughput.

9:58 We didn't have to change anything.

10:00 We literally took the same code that we had optimized for Blackwell.

10:03 Run it on Rubin got 3x improvement.

10:06 Then we are like, can we actually make

10:09 Astra run agents using Astra as the base model?

10:13 Go optimize this inference off of Astra itself on Ruben.

10:17 It's a recursive loop.

10:18 The model is figuring out how to optimize its own inference,

10:22 serving on Rubin chips and over 72 hours.

10:25 The model produced another 2x improvement on the throughput, right?

10:29 So things that would take weeks, if not months to optimize.

10:34 In the past, we now are getting models that are

10:37 very capable of actually producing this over a weekend.

10:40 You just are throwing more compute at it, more black holes at it effectively,

10:45 but I think the larger takeaway point is good hardware plus good tooling, right?

10:50 The reason this happened so quickly over 72 hours is good.

10:55 Programable fact compilers.

10:56 All of those building blocks that DNN team built allow the model

11:00 to quickly explore how best to optimize inference on these chips,

11:05 and they were able to produce a two X benefit over,

11:08 uh, what was already possible because of the hardware improvements itself.

11:12 I mean, our relationship goes back all the way to, um,

11:17 from the first, GPU supercomputers, all the way back to the Ampere generation.

11:22 And, um, it's one thing about opening I they're not shy about giving feedback.

11:28 And so by building, uh, taking that feedback,

11:32 learning and understanding how we can make better tooling, better AI models,

11:36 better architecture to deliver the best possible performance for inference,

11:40 for training, for pre-training,

11:42 for post and for and not to mention just operating it at scale.

11:46 Um, all of that becomes a body of work that, um,

11:50 engineers know how to think about these kinds of machines,

11:53 thinking about them at scale, but also agents or the HCI and the AGI of ASI,

12:00 of the of the interface now becomes something that we can both be optimizing.

12:06 So I'm curious now, what was the piece of feedback from OpenAI

12:10 that you remember most that affected the NVIDIA roadmap in a certain way?

12:14 Well, it was a big, um,

12:16 risk to go from the hopper generation, which was one box.

12:20 We put eight GPUs in one node, and that was kind of the unit of compute was

12:25 a single server to realizing that we're going to trillion parameter models.

12:29 Let me say we talked about it like it was nothing.

12:32 But back then it was like, oh my God, a trillion parameters.

12:34 Um, the only way to get there was to actually disaggregate,

12:37 to take that hopper box with GPUs and to a few NVLink

12:41 switches and now spread out across the entire rack to build 72.

12:45 It was a, um, uh, existential moment to actually do that.

12:49 All of the engineering, all the rest.

12:51 And that, you know, that future of of going

12:55 to these frontier models, OpenAI and others.

12:58 Uh, but but having OpenAI right there with us to, you know,

13:02 the first time we tried to optimizing it to building it and then building it,

13:06 working at 100,000 beyond GPU scale is a, uh, a journey.

13:10 That is that was I think now, at this point,

13:13 Astro is a testament of all of that work that we've done.

13:17 And while your team was working on Astra, on Vera Rubin,

13:20 the whole rest of the AI space was talking about safety and alignment pauses.

13:25 Given what Sam has said about pausing,

13:28 does that impact these aggressive compute build

13:31 out plans that that you've been talking about?

13:34 If anything, I think it actually, uh,

13:37 double underscores the need for more compute.

13:40 The way we will achieve safety and alignment for these very capable

13:45 models is by actually spending and more compute on building the safety models,

13:51 the alignment models that will make sure

13:54 that the frontier models are actually safe and aligned.

13:58 So I think the the perhaps counterintuitive

14:01 conclusion from that is we'll actually be spending

14:04 more compute because not we will obviously

14:06 continue to train and infer these frontier models.

14:09 But in addition, we will be investing in more compute

14:12 to make sure that these models are safe and aligned.

14:15 So it's perhaps a counterintuitive observation, but if anything,

14:18 we believe that we will need more compute to make

14:20 sure that future models are more safe and more aligned.

14:25 Again, um, OpenAI has talked quite a bit about how

14:29 it's used its models and designing its own jalapeno chip.

14:32 Can you talk a little bit about how NVIDIA uses AI models in chip design,

14:37 and is it NVIDIA's own open source models?

14:39 Is it third party?

14:41 We use them all.

14:42 We absolutely use them all.

14:44 Um, some of the open models, our own Nemo, Tron model.

14:48 Absolutely.

14:48 GPT and and others.

14:50 Um, our engineers have access to all of it.

14:53 Uh, and then we're build,

14:54 build our own tools and our own tooling and our own flows.

14:58 A couple of, um, uh,

14:59 I will say our bug databases is dates back to when I was an intern at Nvidia.

15:04 So it is a wealth of information

15:06 that teaches us about our own products, that actually,

15:09 it's wonderful having an agent actually ask you what's actually

15:12 going on with this bug and be able to summarize it.

15:14 And all the engineers in the status.

15:17 Um, we use it.

15:18 So for understanding issues in our hardware,

15:20 we use it for optimizing the silicon yields.

15:23 Um, we actually even use it for code check ins.

15:27 Um, you can't just check your own.

15:29 No one can just check in code to the Cuda driver.

15:31 You have to have it properly reviewed by, uh,

15:34 domain expert or by a peer to make sure there's no bugs.

15:37 It's best practice, but now we have an agenda

15:40 tool that they can actually pre submit to the tool.

15:42 And it looks for common issues or known bugs,

15:45 because we actually have individual agents that are aware

15:49 and knowledgeable about different aspects of our software stack.

15:52 And the engineers really appreciate that because they can they

15:56 can get it checked before it goes to the the expert.

15:59 What's also interesting is that it's a psychological experiment.

16:01 When we first turned it on, we had like 1 to 2 our agent test,

16:05 and all the developers hated it because it was like slowing the down like,

16:09 oh God, I want to just finally die.

16:12 I wanted to review it.

16:13 Um, it was actually, uh, GPT one amazing thing about GPT is

16:17 that it really has really great token efficiency.

16:20 And so we were able we realized we had to bring down this pre-check test.

16:25 And by using GPT, we can do it with fewer tokens.

16:29 It happens faster.

16:30 And for those of you who are maybe doing this for yourself,

16:34 15 minutes seems to be the good.

16:36 Developers are good with a 15 minute tool on a pre-check.

16:39 If you go any longer than that, they start

16:42 to complain just regardless of how useful it is.

16:45 So thank you.

16:46 Um, Ian, if we could go as we're going through your career,

16:49 we just covered off on intern a little bit later.

16:51 Uh, you created Cuda.

16:52 Um, it's now, you know, about two decades on, uh,

16:57 as software systems and chip systems become more and more complex,

17:00 what kinds of new software do we need in networking and orchestration?

17:03 Or perhaps another area to handle that?

17:06 I think, um, as such, and mentioned,

17:08 Astra is an excellent kernel optimization tool.

17:11 I think we the it AI is another tool in our toolbox,

17:15 along with all the other profilers.

17:17 None of these other tools are going to be replaced.

17:20 They're all just complementary,

17:22 and they're all be taking advantage of of AI and computing in various ways,

17:26 just like the cloud, uh, really accelerated the productivity of those tools, uh,

17:30 in terms of what the platform looks like and what how how to engage developers.

17:36 Um, they now have agents to to pull from.

17:39 The agents can be using the tools directly or the, um,

17:43 the libraries themselves can exist.

17:45 Um, every single domain speaks in a different language.

17:49 Sometimes they're even the same algorithms.

17:51 But if you're developing software for, um,

17:54 EDA, if you're a silicon vendor or, um,

17:57 the fluid simulation or an airplane wing or the operational,

18:02 um, logistics of, uh, shipping or delivering pizzas,

18:05 they all have speak in a language of libraries and algorithms and concepts

18:11 that are known both to the users and the agents and the AI.

18:17 Uh, we have over 8000 applications that are been, uh,

18:22 GPU accelerated, major ISV applications that we help track and curate, um,

18:27 the capability and making those developers productive in the end is what's going

18:31 to connect them to the value of what they're bringing to their users.

18:35 So while I don't, um, individually manage through it anymore, uh,

18:39 one of the joys of my job is to help those industries adopt AI,

18:45 adopt computing, and figure out how to take advantage of all these technologies,

18:49 uh, and computing in general to be more.

18:51 And, and such.

18:52 And obviously, you know, OpenAI is running systems with, again,

18:55 a bunch of different semiconductors.

18:57 How do how do you think about orchestration and software

19:00 needed for systems that are in data centers that are increasingly,

19:04 you know, heterogeneous?

19:05 I think we see heterogeneity everywhere, uh,

19:08 was talking about at the start or even for training,

19:11 as well as for products and inference.

19:15 We'll have obviously GPUs, but also CPUs, networking, storage,

19:19 all of which needs to be stitched together and orchestrated,

19:23 increasingly user expectation around interactivity.

19:26 Uh, but that that needs to be a very tightly controlled loop.

19:31 I think overall where we see this evolving is,

19:35 uh, heterogeneity is a fact of life.

19:38 We try to get away from not had to deal with it for so long,

19:42 but I don't think we can anymore, or as we think about the future.

19:47 It's we're a small team.

19:49 OpenAI is still a very small team, relatively speaking.

19:52 And so our push is being on how do we use our own models,

19:55 how to use our own AI to build out this orchestration software.

19:59 And just like with kernel optimization,

20:01 Astra is actually very good at even end to end optimization.

20:04 You can give it an end to end problem,

20:06 and it can figure out how to sort through it and actually make the end

20:11 to end system work So we feel increasingly

20:13 we will build with our partners like NVIDIA,

20:15 the right tooling, the right telemetry, expose the right knobs,

20:19 and really let agents go at it to figure it out.

20:23 End to end.

20:24 Great.

20:24 Well, that's all the time we have.

20:26 Sachin and Ian, thank you for joining us.

20:29 Thank you.

Study with Looplines Download Captions Watch on YouTube