What Open Source Teaches Us About Making AI Better - Ep. 278

0:10Hello, and welcome to the NVIDIA AI Podcast. I'm your host, Noah Kravitz. Right now, the world is watching AI evolve faster than ever before. And that progress isn't just being fueled by technological breakthroughs in scale. It's being fueled by human collaboration. Open source models, open data sets, and shared research are giving developers, enterprises, and governments the building blocks they need to innovate together. NVIDIA has been part of this movement from the very beginning, contributing open libraries, publishing datasets and research, and most recently, sharing families of open models.

0:45Which brings us to today's episode. We're talking about Nematron, specifically, unlocking the secret of Nematron. On the surface, Nematron may look like just another open model family. But the real story is how it anchors NVIDIA's strategy for building accelerated infrastructure and driving increased adoption of AI everywhere. Joining us to unpack this open Secret are two of the leaders driving this work forward. Brian Catanzaro is Vice President of Applied Deep Learning Research at NVIDIA, and Jonathan Cohen is Vice President of Applied Research at NVIDIA. Brian and Jonathan are here today to talk Nemetron.

1:20I can't wait. Gentlemen, welcome to the AI Podcast. Thank you so much for making the time to join us. Thank you for having us. Yeah, it's great to be here. So let's start at the top, and I'll direct this one to you, Brian, to get us going, if that's all right. What is Nemotron? And as a follow-up, why did NVIDIA decide to build its own family of models when you already work with essentially every major model builder out there? Nemotron is NVIDIA's open technology for artificial intelligence. Nemotron includes models that we train. It also includes data sets that we release, as well as algorithms and methodologies.

1:59And our goal with Nemotron is to support the community in building customizable AI that can be integrated deeply and tightly into the beating heart of every business around the world. Our second goal with Nemotron is to help NVIDIA design systems for deploying and constructing AI. There's a lot of questions about how AI works that touch the various design decisions that go into to building NVIDIA's software and hardware systems. And we can answer those questions better because we build Nemotron. So, you know, ultimately we're excited to open up Nemotron even further and continue to put it out there for the community.

2:44We love learning from the community. Nemotron is built in collaboration with the community where we learn a lot from what others are doing in the community. And then we try to, you know, contribute what we can back. We think that this is a great opportunity for NVIDIA to support the AI industry. Yeah, so Nemotron is a collection of large language models. And it's probably worth saying, so they're text models and multimodal LLMs. And we've kind of settled on like three sizes, or we think of them as weight classes. So we have smaller models that we call nano models. We have medium-sized models we call supermodels.

3:22And then we have the largest frontier-sized models. which we call Ultra. So, Nemetron collectively refers to everything Brian said, and then this family of models that we're called. And so, how does Nemetron fit into NVIDIA's broader AI strategy? Because from what I understand, it's not just, and I say just, the models are huge, but not just the models, but it's kind of a cornerstone for growing the ecosystem. Yeah, well, you know, if you think of nvidia as an accelerated computing platform company and you asked the question well what does an accelerated computing platform mean in this age of ai so it includes chips it includes networking it includes software stack but it also includes the models and you know when we think about what is a platform today a platform is all of those components and if you're building ai applications and you you care about the quality of the models but you also care about the performance Like Brian mentioned, one of the reasons that we train Nemotron models is so we can learn.

4:23We are pushing the limits ourselves so that we can learn and make sure that our platform is the best. But it also means we can do co-design. We can cooperatively design the model architecture, the software stack, and the hardware across the whole, all of the hardware components all together. And we've been doing that. And that gives us opportunities to make things more efficient, lower latency, higher throughput, more energy efficient by improving things across that entire stack up into the model architecture. And so Nemotron is a really important part of that strategy as a accelerated computing platform company where our success comes from this full stack co-design and optimization.

5:00Yeah. One thing I wanted to add to that is that these days there are new things that are part of accelerated computing that maybe people haven't considered. So for example, data sets that we use for pre-training and post-training models have a dramatic effect on how quickly the model converges. In fact, comparing different revisions of our Nemotron pre-training set, we've accelerated pre-training by a factor of 4x just by having a smarter pre-training data set, which means that you can actually train a much smarter model with the same amount of compute. Yeah. What makes one data set better, more optimized to help the model converge faster than another?

5:45Well, what we're trying to do with LLMs is build something intelligent that can help us solve problems. It can answer questions. It can reason. It turns out that if you just take all of the texts that humankind or computers have ever produced on the Internet and train an LLM on it, that's kind of where the community started many years ago. But it turns out that's not the most intelligent way of building AI, because a lot of that text isn't adding very much intelligence. And so every organization that builds LLMs spends an enormous amount of effort and compute in understanding their data set, refining it, rephrasing it using synthetic data generation.

6:24and the effort that we put into these data sets has an enormous impact on how quickly the models train and also on the overall strength of the model once it's trained. And so these days, I believe that the data sets that we release as part of Nemotron are an important part of NVIDIA's accelerated computing efforts because it's not really possible to think about how fast a system is for training if you're training on a data set that's not very smart it's going to take an enormous amount of compute to get to the same amount of intelligence as if you were training on a data set that was much more polished.

7:03And so that's kind of the genius of accelerated computing, is that we try to understand the problem from first principles, and we try to optimize the entire stack end to end. And these days, it seems pretty important that NVIDIA's accelerated computing platform includes NemoTron. Right. I can give another example that's interesting. So reasoning, the way these models reason is they generate thinking tokens, right? You ask it a question, and then it generates a lot of tokens as it thinks through the answer. And there's very clear examples where you can generate a lot of tokens and not actually make a lot of progress towards the answer, or you can be more efficient, generate fewer tokens and make more progress.

7:48And again, from the same perspective of accelerated computing, you don't really care, you know, did it generate 10 ,000 tokens? Well, you do care. You care if you can generate, you know, the same quality answer in 2 ,000 tokens instead of 10 ,000 tokens, that's a 5x speed up. And so that's also part of the accelerated computing story. So all of these are opportunities we have to make things faster. Right, exactly. Accelerated computing has never just been about how many arithmetic operations per second you can perform. It's really about what capabilities do you provide. Yeah, yeah. And I think the key to NVIDIA's historic success is, as a company, we've always focused very much and had deep expertise on the actual end applications people care about.

8:30You know, whether it was computer graphics or hyperforms computing, or deep learning, or now modern AI, it's really thinking about what's the end goal and how do you build a platform that gets you that goal with the least amount of time you have to wait, you know, lowest latency, highest to it. You talked some about the openness, the collaboration and co-design that's such an important piece of this. Open source, a big part of Nemetron. What does it mean, and maybe Brian, I'll ask you this first, but either of you, what does it mean to call Nemetron one of the most open AI development efforts that we've ever seen?

9:03Well, we really think that it's important for AI to be trusted and widely deployed. And in order for that to happen, we think it's important that enterprises have the option to understand the data sets and the technologies behind AI and fine-tune them for their own problems and then integrate them very tightly into the software and systems that they use to solve problems for their markets. We think AI is not a one-size-fits-all solution. And we've seen in the past many instances of when open platform technologies really allow different industries to differentiate different solutions for the problems they face.

9:47For example, the internet as an open technology had really different implications for different industries like healthcare versus retail. You know, the way that those organizations use the internet to change the work that they do was quite different. But the fact that the internet was an open technology allowed many companies, many industries to think about solving their problems in a new way using the internet. And when we think about AI, it seems obvious that enterprises need that ability as well. You know, the world's most important and valuable data always has the most sensitivity about it.

10:26And so we think it's important to support enterprises as they learn how to deploy AI, that they can do it in a way that respects their work, their privacy, the important ways that they go about problem solving in sort of a unique way for their business. And so we think that it's really important that there exists an open foundation for organizations around the world to build and deploy AI. And Nemotron is how we're contributing to that. I can just add one thought to that. From the perspective of accelerated computing, if you think about, you know, we come up with some way to make a chip faster, how does the world consume the benefits of that, you know, acceleration?

11:08Well, in the case of a chip, you buy a chip and you get the benefits. But what if we come up with a technique that makes models more efficient at thinking or a data set mix that saves you time and training? How does the rest of the world receive the benefits of that? In what form do you package it? I think the answer, the only answer is we have to teach everyone what we did by sharing it through open source, open weight models, sharing the data sets, explaining how they work, sharing the algorithm. So I think it's natural that open source is a delivery mechanism for the technology that's going into our platform.

11:45So from a little bit of a hypothetical, I'm an IT leader or a business leader at an organization, and I'm hearing what you guys are saying, and we want to do this. And we have specialized needs in our industry, and we have troves of our data that kind of represent company intelligence and our special way of doing things that has brought us success. And we're ready to embrace the AI age and transform. We could use Nemetron, and I'm going to walk through this and point me when I get this wrong. We could use Nemetron to take an open source model and we could customize it, train it on our company data and the rest of the industry data things to help it understand what we do and the problems we're trying to solve in our industry.

12:32Nemotron could help. We could add reasoning capabilities and that sort of thing to it. And then we have a kind of, and I don't want to misuse the term sovereign, and if you guys want to talk about sovereign AI, but we would then have our own sort of customized, adapted to our business, our industry, the way we do things, our data, and it's ours because we took an open model, we trained it, and so we don't have to worry about the sensitive data being out in some commercial model somewhere or what have you because it's our model now. I think that's one aspect. Close. Yeah, that's one. I mean, there's many aspects.

13:06So, for example, if you say, you know, NVIDIA trains a model, a Nemotron model, and it's great. But since you've disclosed all your training data and look at your training data, for whatever reason, we have some policies where this data we can't use. And we can say, that's fine. Everything you need to reproduce what we did is there. You can train your own model, excluding that data. Or you say, well, I like the data, but the mix is wrong. I don't know. I'm a sovereign project. And it really needs to be very good at speaking this language and understanding this culture. and that data wasn't as represented in your training set as I want it to be.

13:39Everything that we did is transparent. And so you can make these modifications yourself. I mean, that's one aspect. Fantastic. Yeah, right. NVIDIA has released datasets, recipes, alignment techniques alongside the models. So along these same lines of building trust and transparency, why is all of that important? Why is this full level of transparency important for the end users, you know, to be able to customize and deploy safely? Well, I think ultimately, if you don't know what's in a technology, it's harder to trust it. And every business has different ways of thinking about the problems they're solving.

14:16They have different problems. and I think it's important as we get more sophisticated about deploying AI and we integrate it more tightly into business problems around the world for businesses to be able to inspect, you know, how is this AI built and, you know, therefore I can build trust that it's going to help my business solve problems. You know, the integration is a really important point as well. So with Nemotron models, there's a really broad spectrum of integration. You can run it locally on a machine without any internet. You could also run it through an API in the cloud and everything in between.

14:54You can deal with your business's sensitive data using the same data management and security protocols that your business already has. And I think for a lot of applications of AI, that level of customizability and introspection is going to be essential. I also want to say that I think there's a real big benefit to open technologies in the sense that they tend to develop faster. So NVIDIA believes that helping AI grow creates opportunity for us. And we think that one of the best ways of helping AI grow is to contribute in an open way to the community. I think when you consider a technology that's being developed kind of independently by a few different organizations, but they're not able to share very much about what they're doing, there's obviously going to be a lot of reinvention that has to happen.

15:49and the progress is going to be slower. And so if we are able as a community to come together, you know, to contribute ideas, data models to each other and learn from each other, I think that will progress faster. And, you know, we've seen that over the past couple of years as various organizations have been contributing to the open technologies for AI. It's really help the community move forward. And, you know, like, for example, OpenAI just released GPT OSS. That was a fantastic thing for the field. Alibaba has been doing some great work with QN models, obviously, Neta's family of llama technologies has been extraordinarily helpful to the field, to help the field grow and develop.

16:36And at NVIDIA, we know that when AI grows, it's opportunity for everyone, it's opportunity for businesses that they can solve new problems. and it's opportunity for us because we work with every business that's building AI. Yeah, I mean, a good example of that playing out is our own research groups often will, like if you have some idea for a way to improve a model, we often will just take one of the existing open-weight models, not necessarily Nemotron, that sort of gives you the best vehicle for trying out your idea, improve it in some way and publish a paper, release a result, right? So like we are building on all the work from these other organizations that release open-weight models all the time as well.

17:14And that's, you know, this is no news to you guys or probably many listeners of the show, but that same sentiment has been echoed so many times over, I mean, over the past couple of years in particular, by guests we've had from all industries and walks of research in life. And you know that the more we're collaborating, the faster we move as a whole. Yeah. Our guests today are Brian Catanzaro and Jonathan Cohen. They're both from NVIDIA. Brian is vice president of Applied Deep Learning Research, while Jonathan serves as Vice President of Applied Research. And they're here talking to us about NVIDIA Nemotron, a family of open models and open technology.

17:52We've been talking about the importance of open, open technologies to the AI community in general, to NVIDIA, the learning that goes into informing really the whole stack, the hardware, the models, the software, the connectivity, networking, everything. And the data sets, as Brian was talking about, and how it all really comes together to make things advance faster and more efficiently, sort of broadly speaking. Neumatron has been a huge effort at NVIDIA with many teams working together, they still are, to bring this to life from advanced research to commercially licensed models and datasets now.

18:27Can you guys talk about the pipeline from research to production models, what that's like, what it's been like for Neumatron? Well, it is a huge effort and it takes a lot of people with different talents coming together to build Nemotron. We've organized the project around basically the different stages of development that a model has to go through, pre-training, post-training, alignment, and so forth, as well as different functional areas, like for example, long context recall or image understanding. So within each of these areas, we have multiple teams working together, some of which are very researchy, very theoretical, and others are very engineering focused and then whole spectrum in between.

19:12I would say it's a great honor to be part of a project where people are coming together to build something like this. It's also a big challenge, you know, trying to get so many brilliant minds pointed in the same direction. uh i think that's one of the central challenges facing every ai development effort around the industry these days is how do we work together to build one amazing thing as opposed to building a hundred small things yeah and you know that's that's really something that's been inspiring to watch uh come together yeah if you compare it with like a large-scale software effort you know there's this like famous uh observation called conway's law which is the the communication patterns that are observed within a piece of software tend to mirror the organization, the communication patterns of the organizational structure that build that software.

20:03And training a model is like, I mean, Conway's laws is definitely an issue, but it's just a very different endeavor. It's not like I build a module and you build a module and we have a nice clean interface. Somehow all of these things have to get combined together. You know, image, Brian's example, image understanding and long context recall. somehow I'll have to get combined together into a single training recipe and a single dataset mix. And so the modularity is, I think, less than in software engineering. And so this idea that you can just decompose it and have lots of teams with sort of clean interfaces between them doesn't really work as well.

20:41And so I think there's a real struggle in scaling up an effort like this to a very large team to do something really big. Is there a new paradigm emerging? It's an interesting question. Organizing, yeah. I wonder, you know, over the next five, 10 years, there'll be some new law named after someone and some new management principle here. It's an interesting thing we've certainly been thinking about. But it does present these challenges. I think one of the most important principles that we've kind of settled on is you just need a lot of internal openness and transparency. You have to solicit ideas.

21:12There are a lot of people across the company and outside of the company working on all these problems. You have to solicit all these ideas and you have to encourage them all to work together. That's the only way forward. And so that just takes a very mature culture and good leadership and ego-less operation and everyone being really motivated at the end of the day by the work. I would say also that one of the amazing things about AI is that it's such a general technology that it really changes the way that we do AI. You know, it used to be like 20 years ago when I was a grad student that it was common for people to build state of the art models in computer vision on their own, like one graduate student on their own to build a model that's that was state of the art in some important area of computer vision.

22:07And, you know, that's kind of how we were trained as PhD students is like, go be brilliant on your own. Well, with modern AI, the best results come from using industrial scale equipment and, you know, general models that can then be taught how to solve important problems. But that requires working together. So one of the first things that AI has changed is the development of AI itself and organizations that can figure out how to collaborate and work together succeed. And, you know, that's one of the reasons also that we really believe in Nemo Tron as an open project is because we've seen how openness internally has made it possible for us to solve whole classes of new problems with AI.

22:55We believe that as Nemotron and other open efforts come together, bring together more ideas and more force to bear on the development of AI, that the results will be stronger. Jonathan, NVIDIA has a history of building end-to-end products. Self-driving comes to mind, gaming, of course, SuperPods, but then disaggregating them for the world to use. Does Nemotron follow that same pattern in your mind, and if so, how? Yeah, I think so. So I think when we talk about that, and Jensen talks about this a lot, what we mean is our solution, but the things ultimately that we build are very complicated, integrated systems with many layers and many components.

23:37And on the one hand, we need to build the whole thing ourselves, because it doesn't work unless you build the whole thing yourself. So we need to train a whole model at the end of the day. But it doesn't make sense for us to release, I don't know, a way to make a reasoning recipe without actually training a model to do reasoning. You have to do these things and put the whole thing together. But at the same time, I think it's very important that we put all of the components into the ecosystem and allow people to consume the parts that they want and not consume the parts that they don't want. So this is how our hardware is.

24:11We design data center scale computers at this point. But we don't sell it as a single data center. We design the whole thing, we build the whole thing, then we chop it up into pieces and we sell it through, you know, normal sales channels and, and people, our customers are free to take the parts they want, replace, you know, it's truly an ecosystem. You know, if you don't like the way you don't like our CPU, use a different CPU, you don't like the storage, use a different storage, you don't like this networking, use a different networking. And we're open and interoperable with all these things.

24:42And it's a tremendous engineering challenge to work that way. But I think it's why we've been so successful is because it allows us to harness the power of the entire computing industry because we're not really locking anyone out at all. We're including everybody. And so when we think about large language models, I guess we're thinking in the same way. So we're going to develop techniques and anyone is free to take them. Other companies that train large language models for a living are free to take anything we built. They probably won't take all of it, but they're free to take anything. They want to take the software.

25:12That's great. They want to take some of our data sets. That's great. They want to take the software and the data sets and some of the training recipes, but modify them. That's great. They want to take the finished models. That's great. So in that sense, I think philosophically, that's absolutely how we think about products, how we think about hardware, how we think about software. And it's now how we also think about foundation models. And I think that's one of the things that makes NVIDIA unique as a big tech company is that although we do full stack and integration, we don't dictate to our customers how that technology is going to be deployed or used.

25:48We know that it's not a one size fits all problem. Or even assembled. Right. And so we're happy to support companies of all shapes and sizes in every industry develop and deploy AI. And because NVIDIA has this orientation, this supportive orientation, where we understand that it's not one-size-fits-all, that actually is the secret to why we are able to collaborate with all of these companies. And we want to do that with AI technology as well. Kind of switching gears a little bit, but still talking along technical lines, can you share any exciting technical breakthroughs that came about during the Pneumatron development process?

26:30and what they might mean going forward, specifically in terms of efficiency and deployment, but really take it as broad as you like. Yeah, well, NVIDIA is thinking about AI from an accelerated computing perspective, and we have a belief that the faster we can make a model, the smarter it's going to be. And this follows just because, clearly, if we're able to think quicker, then we can get more thoughts in the same amount of time that can help us solve problems. So we're bringing this perspective of accelerated computing to AI in kind of a unique way. A couple things just from the past few months that we've demonstrated that I'm really excited about.