Latest / Agentic Conversations (formally mlops.community) / Efficient GPU infrastructure at LinkedIn // Animesh Singh // MLOps Podcast #299
Transcript
- 0:00Hi, this is Animesh Singh. I'm the Director at LinkedIn
- 0:03leading our GPU fleet and infrastructure for training and
- 0:05inferencing. In addition, I lead the
- 0:08distributor training software stack across all of LinkedIn,
- 0:12additionally contributing to a leading a lot of initiatives
- 0:15that are on MO efficiency and performance optimizations.
- 0:20Prior to that, I was a Distinguished Engineer and
- 0:23Senior Director at IBM, leading a lot of core Watson
- 0:27initiatives. Woo, I'm bubbling from this
- 0:31conversation. So many gems when it comes to
- 0:39working with LLMS at scale, GPU infrastructure, what you want to
- 0:44be optimizing for, how you can think about optimizations, and I
- 0:51had. A.
- 0:52Incredible question at the end. Like how does the platform and
- 0:57the GPU infrastructure that you're dealing with differ when
- 1:03it comes to working on MLMS versus traditional ML?
- 1:09Did not disappoint. Let's get into the conversation.
- 1:19Well. Let's start with this, man.
- 1:21And I'm so happy that we got to do this because we've been
- 1:25jiggling around our calendars to make this work for probably
- 1:29about six months. I think we were going to do this
- 1:31when I was last in San Francisco in June.
- 1:33But one thing led to another, and here we are in 2025, having
- 1:39the conversation. Finally, persistence paid off.
- 1:43I think this is great and I've been following your work
- 1:47throughout as well. You're doing excellent work in
- 1:49terms of, you know, bringing communities together and
- 1:52disseminating that knowledge, right?
- 1:54Like what's all happening in the space, right?
- 1:58And what in these cases are springing up?
- 2:00What are the industries they are targeting?
- 2:03There is some excellent work you are driving in that and I'm so
- 2:05glad it's happening in 2025. We have now some better
- 2:11experience of, you know, what is working, what is not working,
- 2:17what may be a little bit of the hype, right, What is realistic,
- 2:20what is going to be the trends in 2025.
- 2:22So I think the timing is is working out, yeah.
- 2:26What is working? One thing which has definitely
- 2:31proven that it's here to stay is NLMS, right?
- 2:36I feel throughout, you know, 20, 2223, there was a lot of, you
- 2:47know, discussions how effective LLMS are going to be right in
- 2:53the industry, in the space. There are a bunch of, you know,
- 2:56modelling architectures like, you know, recommendation ranking
- 3:00LPMS, graph neural networks, GNS, right, But the efficacy of
- 3:06LLMS in the use cases being powered by LLMS was quite a bit
- 3:12of question mark, right? Like, yes, there was a promise.
- 3:14What we saw that magic moment with Chad GBD coming in that
- 3:19literally, you know, woke every up everyone uprightly.
- 3:22It does seem seamless. It it does seem that it's it's
- 3:26not yet another chart what you are talking to.
- 3:29And that sprung up the industry, right when I joined LinkedIn at
- 3:35that time, you know, the chargeability moment hasn't
- 3:37happened. And as soon as I joined, a month
- 3:39later, chargeability came in and what I came in here to do, a lot
- 3:43of that changed within a period of a month.
- 3:46And I think through the course of that period, multiple
- 3:50companies, multiple industries have identified, right,
- 3:53different use cases which are working well with this, right?
- 3:58And people are being productive, be it, you know, either, you
- 4:04know, generating code, be able to do certain automation
- 4:09leveraging this. The interface does seem very
- 4:12common like and a lot of the generative VI use cases which we
- 4:16launched even on LinkedIn, for example, you know, profile
- 4:20summarization, right? So based on what you have, you
- 4:24know, create a headline for me, create a summary for me, use
- 4:28cases like, you know, assisting falling the learning courses,
- 4:33use cases like, you know, resume targeting recruiter emails for
- 4:38candidates. Because a lot of the things
- 4:40which we had seen that, you know, if you're getting cold
- 4:43call emails from recruiters, they are not hyper personalized,
- 4:47right? Like at times they sound like a
- 4:50template generative AI in LLMS helped us immensely in that
- 4:54right where they take into account the candidates profile,
- 4:59the company in which candidate is working, the company from
- 5:01which the recruiter is and create these very personalized
- 5:05emails which you know, we are seeing success candidates
- 5:08actually this morning much more opening those emails.
- 5:12So I think you know, a bunch of use cases we have seen working
- 5:15really well with LLMS and we are obviously doubling down, right?
- 5:20Like if you see the talk of 2025, anyone before that last
- 5:25year there was quite a bit of discussion on agents, what they
- 5:30can do, what they cannot do. And we did our own experiments,
- 5:36right, like invested a lot internally in terms of building
- 5:42agent infrastructure, right? First of all, right, what does
- 5:47it take to create agents? How's it different than the
- 5:50traditional generative AI applications or use cases?
- 5:54We were building more than once is right?
- 5:56What makes it different? And then finally, we launched,
- 5:59you know, closer in the last quarter of last year, LinkedIn
- 6:04hiding assistant, right? So which is essentially an agent
- 6:07for the recruiters, which based on certain criterias they define
- 6:11will actually go work behind the scenes, find candidates for
- 6:15them, relevant candidates, summarize their experience and
- 6:19profile to the recruiters. And then based on that they can
- 6:22tell, OK, you know, reach out to this particular candidates and
- 6:27you know, let's start having a discussion.
- 6:29And there is much more. We are doubling down on that
- 6:31whole LinkedIn hiring assistant and we have seen some great,
- 6:34great enthusiasm from our partners and customers, right.
- 6:38So it's it's seeing some very good results.
- 6:41So there is much more we are doing now on other areas off
- 6:44LinkedIn which will be powered by agents.
- 6:46So they are here to stay. I think it's how in which use
- 6:52cases really benefit from it. That will be a nuanced
- 6:55discussion. You cannot throw it at every
- 6:56single thing. There are so many things which
- 7:00can be just powered by them, which then frees you up to do
- 7:04things right, which are probably the more creative aspects of the
- 7:08work you are doing right And and we are seeing quite a bit of
- 7:11that happening. So what's not working?
- 7:14I, I wouldn't say what's not working, but a thing which
- 7:18probably needs a lot of improvement moving forward is
- 7:24the cost and the ROI of launching these LLM based,
- 7:31either agentic use cases or traditional, you know,
- 7:35prescriptive rag based generative AIE apps or even, you
- 7:40know, leveraging LLMS for use cases beyond generative AIE.
- 7:45The cost is a big hindrance. I think 1 if you see like
- 7:50training itself used to be the biggest barrier for entry and
- 7:55that God saw to some extent because first of all, you know,
- 8:00even within LinkedIn we invested quite heavily in building out
- 8:04our scale out training infrastructure which can power
- 8:06elements. I think when I came in, you
- 8:09know, we were working off we 100 fleet and then since then we
- 8:14have scaled our LinkedIn fleet by 7X.
- 8:17We have a one hundreds, we have H1 hundreds, we have H2 hundreds
- 8:21and and the fleet is as modern as it could be and we have
- 8:25scaled our training tremendously, right.
- 8:27Like it's 150X increase in the the size of the models we are
- 8:32training, right. Our, our data processing has
- 8:35increased many Ford, we actually, you know, completed
- 8:401,000,000 training runs on the platform and we're training big
- 8:44foundation models now investment in the infrastructure renting
- 8:48and I think it was well understood for companies at
- 8:52scale of LinkedIn etcetera, right, where a lot of content
- 8:56based data is being consumed, being produced, you will have
- 9:00tons of data, right? And when you have these tons of
- 9:03data, you need to make sure that you have the infrastructure to
- 9:05train models on these data, right.
- 9:07And then the other thing which actually helped a lot of the the
- 9:12training landscape is the emotions of open source models.
- 9:15I think, you know, Meta led the way, followed by many of those
- 9:20in the industry, right? Where for smaller companies or
- 9:24for companies, which there is not a need to train a model on
- 9:28word data, right? Like you have your own specific
- 9:31data, which is probably, you know, not as big as as the word
- 9:35data. Getting these open source models
- 9:38as starting on top of them, because these models already
- 9:41know what exists in the world. They can answer your questions
- 9:44around, you know, they've gone through Wikipedia, they've
- 9:47called through, you know, public libraries, all the articles and,
- 9:50and they're being trained on them.
- 9:51Then you can bring in and do more fine tuning instead of, you
- 9:55know, training on huge amounts of data.
- 9:59So then the infrastructure cost can go down further, right?
- 10:03So fine tuning became a big mechanism.
- 10:05Plus, you know, the mergers in the industry around a lot of
- 10:08techniques around supervised fine tuning. 0 short, few short
- 10:11learning which emerge where, you know, prompt optimization
- 10:15techniques which emerge which essentially brought the cost
- 10:19down heavily on the training side.
- 10:23What is now, you know, more and much more in the picture is the
- 10:27cost of inferencing, right? And, and I think it's, it's
- 10:32humongous, it's big and there are a lot of efforts, right,
- 10:35which we are doing, which the industry is also doing right.
- 10:38Overall, how to bring down the cost of inferencing these
- 10:40models? Now if you take a look at use
- 10:45cases specifically as well, there is some thinking which is
- 10:49built in like when you are interacting with model, right
- 10:55where you are asking certain queries, asking you to analyze,
- 10:58you're prepared mentally, it will take some time, it will
- 11:00think through it, right? Even with the emergence of you
- 11:03know, the latest open AI models, right, where there is a lot of
- 11:07reasoning going on, right? So there is a lot of it is
- 11:09actually analyzing its own output, then you know refining
- 11:13it's own output, then the second output is further analyzed.
- 11:15There is a lot of back and forth reasoning.
- 11:16So there are multiple inferencing calls happening.
- 11:19And I think as a consumer we are prepared if we are going into a
- 11:22scenario in this particular context, the model may have some
- 11:26thinking time specifically if you are interacting with the
- 11:28latest open AI models etcetera, then you know that you know
- 11:33you're all asking complex queries which need that
- 11:36analysis. So the latency you are are
- 11:39willing to tolerate now the the argument and, and even to get to
- 11:45that latency, right? Like there is tons of
- 11:47infrastructure investment which has been made, right.
- 11:51The general realization is that, you know, with even with all
- 11:56these investments, we are not able to get our GPU's to perform
- 12:01at maximum utilization, right? Like inferencing is becoming
- 12:05very costly because you are optimizing a lot for latency
- 12:08through port and you know, you have a lot of failover
- 12:13mechanisms which you need to build like almost any company
- 12:16needs to account for hey, I, I have two or three data centers,
- 12:20right? Like if one data center goes
- 12:22down. So there is a lot of redundancy
- 12:24you need to build for applications which are user
- 12:28facing. That means you know the cost of
- 12:30GPU is ballooning, right? And and that's essentially is,
- 12:35is an infrastructure problem which needs to be solved
- 12:39specifically. Now when we do take elements in
- 12:42use cases where the consumer appetite for latency might not
- 12:45be there at all, right? Like there are efforts happening
- 12:51across the industry, right? Like, hey, the traditional
- 12:54Rexis, right, recommendation ranking, like so for example,
- 12:58you go to social media sites, you, you get recommendations,
- 13:01you get feed, you get people you want to connect with all these
- 13:05things like, you know, as soon as you go to the site, this
- 13:09should be there, right? You should be as you're
- 13:12scrolling through the feed, the feed should be just mean, you
- 13:15know, updated and and customized for you in real time.
- 13:18There is no appetite for latency in those scenarios.
- 13:21Now, if, if you, if you need to see if elements can be effective
- 13:26in those world, that means, you know, you nearly really need to
- 13:30optimize for latency. And if you're really doing that,
- 13:32you are potentially throwing more money at the problem.
- 13:36And, and so that problem, I think, you know, how do we take
- 13:40this large language model architecture, the transformer
- 13:43architecture and make it really, really optimized for
- 13:46inferencing, right? Is is becoming a big thing
- 13:48right? Which needs to be solved full
- 13:50scale, right? Do you feel like it's a bit of
- 13:55trying to fit a round peg in a square hole?
- 13:58Because when you throw LLMS at like a Rexis problem, just
- 14:03because we've been doing Rexis for a while and we figured out
- 14:09how to make it real time, why do we need to add an LLM on top of
- 14:13it? It almost in my eyes is over
- 14:17complicating things just to try and use a shiny new tool.
- 14:20But maybe you've seen there's better performance, there's
- 14:23better personalization there or something that I haven't?
- 14:27I think I would more than speaking on my behalf, right?
- 14:33Like in general, there are like, you know, research papers,
- 14:37emerging companies are trying that now.
- 14:40Why would you try something like that, right?
- 14:42I mean, and that's a fair question, right?
- 14:43Like Rexis is already well established, right?
- 14:46And as as a architectural pattern, right?
- 14:48Like the traditional recommendation ranking models,
- 14:51retrieval models including, you know something like graph neural
- 14:56networks and GNS, etcetera, right.
- 14:57They do a very, very solid job at at this and you have seen
- 15:01companies like. And it's fast.
- 15:05It's fast and and like you, you take a look at TikTok, often
- 15:11times you know, the recommendation algorithm is, is
- 15:14is talked about. So you're fairly right like hey,
- 15:17is why is is that why I think there are couple of things in
- 15:22heuristic, right. So the way we have solved
- 15:26recommendation ranking problems in the industry is obviously you
- 15:29have created models, a lot of the companies have smaller
- 15:34models, right. These are not traditional
- 15:36foundation models which have been trained on word data
- 15:38etcetera, right. They have potentially not seen a
- 15:42lot of user interactions and patterns.
- 15:44So then you know, you add things like real trying training,
- 15:50there's a lot of data being ingested in real time.
- 15:52Online training is happening. You know, there is there is a
- 15:55lot of feature ingestion which is happening in real time.
- 15:59What is user interacting with? So there is this new paradigm,
- 16:03OK, which is these models, the LLM models, these are foundation
- 16:07models, they have potentially seen maybe 95% of the patterns.
- 16:12So maybe the what you need to do in real time to update these
- 16:16models is is probably a lesser investment, right?
- 16:20Models have seen majority of the patterns and and if you feed
- 16:23what the user is done right like they, they would be able to
- 16:27predict much more comprehensively.
- 16:29You don't need to do a lot of online training etcetera,
- 16:31writing your time. That's one of the the you know,
- 16:34thinking. The other thing is the
- 16:35simplification of the architecture for something like
- 16:40GN NS, right? What we need to do is you know
- 16:44when you are doing the training GNS, all the data is in the
- 16:46graph format, graph structure format, right?
- 16:49You need to traverse the nodes and the edges in real time,
- 16:52right? Because you don't know pre hand,
- 16:54right? Like how much data you will be
- 16:56processing, which will be the right node, which edge you need
- 16:58to traverse, right? So there is no data pre
- 17:01processing happening and there is a different architecture and
- 17:06it's inherently hard to scale GNS beyond a certain limit
- 17:09because there is live data processing happening while the
- 17:12training is going on or while inferencing is going on, right.
- 17:17So an architecture and then GNN is one example.
- 17:20There are different, you know, recommendation ranking
- 17:22architectures and then companies would have a proliferation of
- 17:26these recommendation ranking models, right?
- 17:27Like every team would create for each of their use cases, they
- 17:32would start from scratch and and create a model which is
- 17:35potentially small model. It does very targeted job.
- 17:37It does a really good job at it, right and then build all these
- 17:40things. So you have bespokes models with
- 17:42different architectures. There are huge number of them if
- 17:47you take the LM route, right, like the the trend which is
- 17:51emerging in the industry right here.
- 17:52If I do create a giant foundation model, right?
- 17:55And and if you, you're obviously very present in what's
- 17:59happening, distillation as a technique is becoming very
- 18:03prominent in the industry. Like I will create smaller
- 18:05models for inferencing, but I will distill it from this giant
- 18:09foundation model. So what you have done is you
- 18:12know, you have sort of centralized the, the, the
- 18:15creation of models in, in potentially a central place.
- 18:21So you can think of a scenario in future, right?
- 18:23Like there is one central team instead of having every
- 18:27different use case in every different vertical within your
- 18:30or creating your own models for their own use cases, which are
- 18:32very targeted. 1 central team, which is, you know, the holder
- 18:36of your organization's data, which is curating that whole
- 18:39data and creating, you know, maybe one or two or three.
- 18:43OK, very simplistic scenario. You're creating 2 giant
- 18:46foundation models, one for generative AI use cases, one for
- 18:48non generative AI use cases. And then you know, the models
- 18:52for all particular use cases are being disturbed from this.
- 18:56So you have simplified the, the overall architecture.
- 19:00You are essentially worrying about the compliance majorly
- 19:04like there is compliance is a big thing as well as you are
- 19:07seeing right there was the the DMA act, there is AI act, there
- 19:12will be other acts that will be coming.
- 19:14If you if you curtail the surface area, how many models
- 19:18you have, right, what data it was trained on, right, then
- 19:22you're not worrying about, you know, the many hundred other
- 19:27models which your organization has created and make sure
- 19:29everyone is compliant. What data it was trained on was
- 19:32that data compliant? You have sort of centralized and
- 19:35simplified that problem. Hiding becomes easy, right?
- 19:38Right Now you do tend to hire like when when I'm looking at
- 19:44certain use cases, like, you know, GNS people who can run GNS
- 19:49scale, right. You need a very targeted skill
- 19:51set. People who understand you know
- 19:52how to target graph data, and then you need to go into the
- 19:55depths of the GPU's architecture where is.
- 20:00Because what a lot of data transfer is happening during
- 20:02that training. So OK is Nvilink good enough?
- 20:05How much HBM memory I have? How much on disk memory I have
- 20:08on on the GPU nodes? Then you know what is the
- 20:11network bandwidth. There is a lot of in depth GPU
- 20:15knowledge you will require right just to go there and and start
- 20:18solving. Plus you know the graph
- 20:19traversal. What algorithms do I need to
- 20:21introduce to? So your sourcing of skills and
- 20:25talent also become simplified. So there is that whole value
- 20:28proposition which could be achieved provided if LLMS do
- 20:34prove themselves. I think many are are, are
- 20:38looking at this problem space and, and figuring that out.
- 20:41So it's yet not solved. And as I said, like, you know,
- 20:45like generative AI as well, this is a problem of scale, like
- 20:52generative use cases. You have to explicitly as a user
- 20:55go and invoke that, right? You will go like I'm giving a
- 20:59LinkedIn example, right? You will go and say, hey,
- 21:02summarize my profile. Or a recruiter will say, launch
- 21:05this particular agent or are you, you know, LinkedIn learning
- 21:10user will go and say, summarize this course for me or explain to
- 21:13me this nuance. So these are, you know, discrete
- 21:16transactions, users going to feed, logging into feed,
- 21:21browsing through the content. This is happening all the time.
- 21:24It is happening at scale, right? So, so then that's a bigger,
- 21:28much bigger scale problem which needs to be solved with LMS
- 21:33overall, right, in a cost effective manner.
- 21:35I think it's it's and very, very, you know, latency
- 21:40sensitive manner. So that's that's the thing,
- 21:42right, which needs to be happening.
- 21:43And we'll see whether they do prove efficacy, like, you know,
- 21:48based on all the results you see, they're able to pass a lot
- 21:51of bar exams, PhD exams, math exams, right?
- 21:54So you'll see, hey, they are intelligent, right?
- 21:56Like, can they be very intelligent for this specific
- 21:59set of targeted problems? I hadn't thought about the
- 22:02simplicity in the architecture and also the simplicity in being
- 22:07able to attract talent that understands this architecture
- 22:12because it is more simplified. And you were talking about a,
- 22:17the cost of having GPUs at scale.
- 22:20And when you have this many GPUs and you're trying to utilize
- 22:25them, you don't want to have any percentage going idle.
- 22:31I imagine you think about that a lot.
- 22:33You're thinking, wow, we're burning money just letting this
- 22:36GPU sit around and we're not utilizing it to its maximum
- 22:39capacity. It does.
- 22:42Is that what you're trying to do, or is that what Liger
- 22:46kernels are trying to help out with?
- 22:49Can you explain that? Any, any infra and platform team
- 22:53which is an Emily infra at this point you would talk, this is a
- 22:56burning thing, right? You have on one hand, you know,
- 23:04you cannot run these LMS, for example, or or these modern
- 23:09recommendation ranking models without GPUs, right training
- 23:14definitely right. And even with influencing now
- 23:18with elements, you have to go on GPUs, right?
- 23:21Like the gender trend in the industry, like if you're going
- 23:25two years ago was here, we will train on GPUs would sell on
- 23:28CPUs, right? That's all.
- 23:30Most of the companies are architected and CPUs were not
- 23:34that expensive. The modern architecture, it
- 23:37doesn't lend you to that, right? So we need to get.
- 23:39So the whole investment in GPU efficiency becomes very, very
- 23:43paramount. And it starts from at every
- 23:47single layer like in our case, like, OK, there is the general
- 23:50thing, right, which also happens in companies right here, you
- 23:52need to allocate a maintenance budget.
- 23:54Now you can be much more generous with CPUs, right, in
- 23:58terms of maintenance budgets, how much spare capacity do you
- 24:02allocate so that if you're 20% of your fleet is being
- 24:05maintained on a regular basis. So that's the delta you need.
- 24:09If you need high availability, you need to spread it across
- 24:11three data centers. OK, let's go ahead and do it
- 24:14right. Those decisions were much
- 24:16easier. They're becoming much more
- 24:17harder when you have GPU's in house because yes, there are
- 24:21certain companies which are potentially invested a lot,
- 24:25right? And and they're deep coffers,
- 24:27but a lot of the companies want to be cost conscious about it,
- 24:31right, like you are. So every decision, Hey, even to
- 24:35the point like looking at certain use cases, what would be
- 24:39the right maintenance budget, right?
- 24:41Like if if you have, let's say you know, in in one maintenance
- 24:45own thousand GPUs, can I allocate 100 just for
- 24:49maintenance, right? Like then you do the cost and
- 24:51it's like, hey, there's a lot of wastage which is going on you.
- 24:53Start sweating. You start start.
- 24:55Thinking about how much money that is and.
- 24:57Yeah. And so, so how do you first of
- 25:01all, start then looking at the workload so that they become
- 25:05more resilient to maintenance? Like so, OK, if I cannot take
- 25:09the previous, the previous approach used to be like, once I
- 25:11know 20% of my fleet is going to be maintained, I will not
- 25:15schedule anything, right? We don't need to worry too much
- 25:17about, you know, for the next 24 hours, even if they're empty.
- 25:21Now you're worrying about that. I cannot leave 200 GPUs lying
- 25:25empty for 24 hours, right? Just because you know, the
- 25:28maintenance is going to happen. So then you look at your
- 25:30workload, OK, how can I make the workloads more resilient?
- 25:34Now if you look at distributor training workloads like majority
- 25:38in the industry are gang scheduled, right?
- 25:40So you do gang scheduling, you know beforehand this distributed
- 25:43training workload will attack X amount of GPU's, right?
- 25:47And there is a lot of you know, synchronization happening as the
- 25:51distributed training is going on.
- 25:52And if if you know few of the nodes fail, you will have to you
- 25:58know, put it back in the queue. Like you obviously need to have
- 26:01checkpoint restore, put it back in the queue, do again gang
- 26:04scheduling so that you know it can resume.
- 26:09So then you look at this workload, OK, maybe in this
- 26:11world we have to rethink the whole lotion of gang scheduling,
- 26:14right? Do we need to build more
- 26:18elasticity, right? Like if if a workload has been
- 26:21launched, 100 GPUs and 30 GPUs are are going to be taken away,
- 26:25can it shrink? Can it expand on its own right
- 26:28and continue going Same for inferencing.
- 26:30I think the need for scale down, scale up is becoming big.
- 26:36Serverless architecture has to become a prominent scheme in the
- 26:39thing of things like you cannot optimize for Max traffic like
- 26:45and and a lot of us are are doing that.
- 26:47A lot of the industry is doing that where you put what is the
- 26:50peak traffic you get and that's your capacity.
- 26:53But if out of 24 hours, 2 hours is your peak traffic for those
- 26:57less of the 20 rest of the 22 hours these GPUs are sitting
- 27:00idle because they are being provisioned to handle their peak
- 27:03traffic. Again, a lot of vested.
- 27:05So, so many of these decisions are now becoming very prominent.
- 27:09So I feel the the emergence of elastic architecture, serverless
- 27:13architecture to handle workloads which which, you know, should be
- 27:19able to scale down, scale up as the capacity is, you know,
- 27:23shrinking, expanding. This is, you know, a big
- 27:27investment area and something like which, which we are doing
- 27:31like, and, and I'm, I'm assuming like, you know, a lot of people
- 27:34are, are solving this problem at different means and mechanisms,
- 27:38right? So yeah, that's that's
- 27:40essentially the the key characteristic there.
- 27:44Well, and you haven't even mentioned the reliability of the
- 27:48GPUs because they're notoriously unreliable.
- 27:50They are like there are, I mean all these NVIDIA GPUs you get
- 27:54for example, right? And, and one of the decisions we
- 27:57took early on and I think for the simplicity part of it,
- 28:05right, like, hey, let's start at least on the training side,
- 28:08right? We'll go with NVIDIA.
- 28:10And so far we have standardized on NVIDIA throughout, but then
- 28:15you know, you have a lot of these checks things which can go
- 28:20wrong because it's just taking a distributed training example.
- 28:24When you are running at scale, you are touching your, in our
- 28:28case, you know, we have Kubernetes as the computing for
- 28:30layer, right? So you have Kubernetes, you have
- 28:33the network, you have the hardware, you have the storage
- 28:36training is, you know, reading and pumping a lot of data
- 28:39throughout any of these points going down is a weak link in
- 28:44addition to the GPUs itself, right?
- 28:47And act with the planned maintenance.
- 28:49So, so, and, and sometimes, you know, for foundation models,
- 28:52your training can run for one week, two week, three-week at
- 28:55stretch, right? When you are actually running
- 28:57training for these large foundation models, you're
- 28:59guaranteed to have some part of the infrastructure having some
- 29:02problem. Sometimes it will be the
- 29:04messaging queue, sometimes it will be the storage system,
- 29:06sometimes it's the the something layer or sometimes it could be
- 29:10just the GP node itself failing health checks ECC errors.
- 29:15Nickel is a problem InfiniBand of the GP US right, which is
- 29:18connecting all this. So this this poses a huge
- 29:23challenge right in in terms of ensuring because these are so
- 29:26long running jobs. How do you make sure that you
- 29:29are being effective, right? And that combined with the
- 29:32problem of gang scheduling because him they need to go all
- 29:35together at once or otherwise they will need to wait if a if a
- 29:38training job needs 500 GPUs, it will be in the queue until those
- 29:41500 GPUs are free, right? You cannot launch it before that
- 29:45because you know you have not I mean, that's how the the the
- 29:51architectures around gang scheduling are built.
- 29:53So this this is a big thing, right, which we have invested a
- 29:59lot in terms of building, you know, very fast checkpointing
- 30:03restore because you assume failure is happening,
- 30:05maintenance is happening. So a lot of investment has gone
- 30:08in from outside in terms of ensuring that, you know, there
- 30:10is automated checkpointing restore which could be done and
- 30:15which is, you know, hierarchical.
- 30:16So we first checkpointed memory, then distributed those
- 30:18checkpoints to some async block storage, then you have to read
- 30:22right and and then you know, there is also discussions around
- 30:25whether these checkpoints can reside on the GPUSSDZ.
- 30:28So because there is a cost of, you know, reading once you have
- 30:32like the training can continue checkpointing in memory.
- 30:35But once you have send it to some remote storage, a block
- 30:37storage or a file based storage, then when you're reading, if
- 30:40something has gone wrong, that takes time, you have to bring
- 30:43that through the network and and bigger models will have you
- 30:47know, many gigabytes of checkpoints, right, bringing
- 30:49them through the network. So then OK, can we leverage the
- 30:52GPUSSDS under the covers itself, right?
- 30:55GPUSSDS obviously have some limited capacity.
- 30:58How much can you store? So bunch of those, you know,
- 31:02architectural elements need to be crafted.
- 31:05So we are investing quite a lot and quite heavily into that
- 31:09Liger you talked about right? Was the other part of the
- 31:12efficiency problem, right, Which is essentially, as we went on
- 31:16this journey, we saw like beyond the, the reliability and, and
- 31:22the scalability, a lot of the use cases, as the model started
- 31:24getting more complex, more bigger, and we had an X amount
- 31:28of GPU fleet, we started seeing a lot of pulls from our
- 31:34customer. Hey, I'm IDE GPUs, I'm not
- 31:37getting GPUs. All my training is running for
- 31:40like, you know, so long, right? And, and I really want to get it
- 31:44done faster. Multiple of these use cases and,
- 31:47and literally people were because, you know, GPUs was our,
- 31:52our scars entity, at least, you know, a year ago, we, we had
- 31:55quite a bit of that, right? We were, as we were scaling the
- 31:58fleet, right? And we were ordering NVIDIA has
- 32:00its own supply chain and timeline, right?
- 32:03The use cases were just springing out.
- 32:05So the we looked at the problem, OK and and you follow the
- 32:08traditional methodology of data parallelism, model parallelism,
- 32:12let's introduce tensor parallelism.
- 32:13We invested in technologies like 00 plus plus because our a 100
- 32:18fleet had very constrained network bandwidth.
- 32:21So how you can scale up training even with that constrained
- 32:25bandwidth? 0 plus plus helped in that.
- 32:28So after we have exhausted every single option, then OK, what can
- 32:33we do? So as we started going deeper,
- 32:36one of the things which we used to do infrequently for our
- 32:40customers was, you know, rewriting CUDA kernels, right?
- 32:43Like a lot of the use cases which are very sensitive to that
- 32:48the training should complete within X number of hours
- 32:51etcetera. When we looked at the the model
- 32:54training code which our models used to produce, there was a lot
- 32:58of improvement. So at times we will go and
- 32:59rewrite the CUDA kernels few certain operations, right?
- 33:03No, you, you bring this to the LLM.
- 33:06The thought which we had was a, we can do this on a model by
- 33:09model basis, but this is not scalable, right?
- 33:13Like there are multiple users coming different models.
- 33:18So the thought, OK, how can we make what what exists to
- 33:22actually solve this problem, which can be a little bit more
- 33:25scalable. So that's where, you know,
- 33:27fortunately at that point, Open AI has launched a Triton, which
- 33:31was essentially a Pythonic programming interface, right,
- 33:35where you can do CUDA kernel programming at a much abstracted
- 33:39layer. Most of our users would be, you
- 33:42know, familiar with Python. So using that.
- 33:45Now the second thing was like, you know, GPU memory is
- 33:47hierarchical, right? Like you have the DRAM, then you
- 33:52have the the HBM memory, then you have SRAM, the streaming
- 33:56multiprocessor, SSMS and GPUs, they interact mostly with SRAM,
- 34:01right, which is very little. So every time.
- 34:04And so all these multiple kernels you have and multiple
- 34:09operators you have in your training code, there's a lot of
- 34:12data transfer happening right from CPU to to HBM to SRAM.
- 34:15Like that is your biggest bottleneck, right?
- 34:18Even though the GPUs can massively parallelize the amount
- 34:22of IO data IO which is happening between different hierarchies of
- 34:25memory, that's becoming a big bottleneck for the training
- 34:28time. So that's when we thought, OK,
- 34:30let's combine what trade on is brought to the table.
- 34:35Let's take this problem and create certain first of all
- 34:39custom kernels for our distributed training workloads.
- 34:42And we started seeing huge gains, right?
- 34:43Like in one of the cases, like just three, 100% increase in, in
- 34:46the overall efficiency majority of the cases like more than 50%
- 34:50decrease in the memory which was being used.
- 34:53What we did was we started fusing kernels, right?
- 34:56Like a certain operations, they can be just combined fused
- 35:00together. Operators can be fused together.
- 35:02You don't need to have 5 kernels to do this different, five
- 35:05different operators, right. And there is some human judgment
- 35:07involved in that to, to make that decision, which, and we
- 35:12have also seen flash attention in the industry, sort of saw
- 35:14that quite a bit, right? And, and it became very popular
- 35:18and it was built on the principles of kernel fusion.
- 35:21So we took that and combined that and it's, it's sold and
- 35:26helped immensely. A lot of our internal developers
- 35:30huge decrease in training time because memory efficiency got
- 35:34much bigger. And once we realized, hey, this
- 35:37looks good and, and let's just open sources.
- 35:40There was not a lot of planning. There was not a lot of thought.
- 35:44It was also like, hey, the all of open source models are coming
- 35:48in and and maybe you know, if community likes it, they will
- 35:52potentially create kernels for those open source models.
- 35:54And if we end up consuming it, it eventually also benefits us,
- 35:57right? Yeah, it just took off, right?
- 36:00I think multiple companies were in the same position, right?
- 36:05It's it's when you like go and put a problem out.
- 36:09They were all, you know, getting this this big load of generative
- 36:12use cases. Lot of LMSGPU crunch was only
- 36:16present and almost everybody right needed to solve this
- 36:22problem for distributed training right at at scale.
- 36:25And and we got a solid reception right in the community right
- 36:29from Andre hugging face. We one thing which we did spend
- 36:35some time on making the use very easy, right.
- 36:37So we integrated with the Hugging Face ecosystem right off
- 36:40the back right etcetera. So and we just last week it
- 36:46completed more than 1,000,000 downloads.
- 36:50Right. This is amazing like we never
- 36:53thought you know this will go all the way there right So and
- 36:57we have got insert lots of good feedback from from the users who
- 37:00are benefiting from it. Yeah, It's funny that you
- 37:03mentioned the memory aspect because we had a guest on here a
- 37:08few months ago talking about how his whole thesis was that we are
- 37:13memory constraint, we're not GPU constraint.
- 37:16GPU is almost like not the way to look at it.
- 37:19It should be really looking at memory and that is the
- 37:22bottleneck right now. And, and I think very directly
- 37:27so right Lexi, ultimately ML is a data processing problem right
- 37:33now. So that does mean that, you
- 37:36know, there's a lot of data moving in between these GPU's
- 37:41right now. The fastest you can process data
- 37:44is if the data is in the memory right now.
- 37:47You go through the hierarchical thing OK GPUHBMS, RAM.
- 37:52The second thing is OK the the CPU on system memory of the CPU
- 37:57on on the GPU nodes. Then you go OK, I I will go
- 38:01outside the GPU nodes right. Maybe the GPU SSD is right
- 38:04itself. But the more you can keep in
- 38:07memory, the faster you can process more.
- 38:10So for LM oriented workloads, workloads because if you look
- 38:16at, you know, people want to bigger context length, right?
- 38:20Even when you are training the model, right, if you increase
- 38:22the context length, you potentially can train bigger
- 38:25models inferencing time, right? Like the amount of information
- 38:30users are sending, right for the model to process, the amount of
- 38:35information the models are generating.
- 38:37Like see Rexis output used to be very straightforward, right?
- 38:42Like it's Rexis recommends you whether this is a good content
- 38:47to show or not in case of generative AID output is also
- 38:51huge, right? So a lot of that data processing
- 38:54is going back and forth. And the more you can, you know,
- 38:57leverage the memory effectively, the biggest gain you can get.
- 39:01So it's, it's rightly so there's a lot of even during inferencing
- 39:05time like KV cache is becoming a big technique, right?
- 39:08Like where you can reuse KV cache as the inferencing calls
- 39:12are coming right for sequence of tokens acceptor right which we
- 39:15have seen before you calculated I.
- 39:17Can save a lot of costs too. Yeah, yeah, you've collect,
- 39:20you've already calculated attention scores for all these
- 39:22different, you know, tokens you are swinging the sequences you
- 39:24are seeing, right? So why do this again?
- 39:27So let's keep it in memory, But how much can you keep in memory?
- 39:29The other part is like, you know, if you start leveraging
- 39:32GPU memory for KV caching, then model itself needs GPU memory
- 39:36for its own processing. So what is that route right
- 39:38boundary and, and most of the H1 hundreds, right, which is now
- 39:43dominantly like if you, if you look at what the industry has
- 39:49probably most in bulk is right now H1 hundreds, right?
- 39:52Or to some extent H2 hundreds. And and 80 gig memory per GPU
- 39:58is, is turning out to be low, right, based on the model of
- 40:02data you're processing. And I think part of it is also,
- 40:05as the use cases emerge, that that starts becoming clearer.
- 40:10Because the way maybe the the whatever went into the decision
- 40:15making was a disk memories for GPU processing whatever GPU is
- 40:19computing. When people start leveraging
- 40:21this memory as a cache, like carry cache like in case of GN
- 40:24NS. Also, we are keeping some part
- 40:27of the graph structure in memory.
- 40:28When you start using it as a data storage mechanism in
- 40:31addition to being used for computing needs or of of your
- 40:35model itself, then it's less than you need more.
- 40:38So I'm assuming you know the the that that realization is already
- 40:45like if you see the Grace Hopper architecture and the Grace Black
- 40:49architecture where they're combining ARM CPUs with the GPU
- 40:53nodes. And then they're creating a very
- 40:55high bandwidth data transfer link between that the CPU and
- 40:59the GPU. So that you can then leverage
- 41:01the the more memory on on the other side of the CPU nodes
- 41:04because there is a very high bandwidth data transfer were
- 41:07realization to this particular use case which is emerging from
- 41:13generative AI, right. It's yeah.
- 41:17I want to give a shout out to the person who who said that to
- 41:19me because I was trying to remember their name.
- 41:21It was Bernie Wu was the one who said that we are memory
- 41:25constrained and you just went very deep in that and makes a
- 41:28lot of sense. Everything that you're saying.
- 41:30You did mention one thing that you worked on quite heavily too,
- 41:34which I've also heard is in absolute pain around checkpoints
- 41:40and speeding up the checkpoints because a, it's not clear when
- 41:45things are going to fail. So you don't know if you need to
- 41:47checkpoint every second or every 5 seconds or every 5 minutes or
- 41:51every day. And so you, if you over optimize
- 41:55for checkpointing all the time, then you're potentially
- 41:57transferring around a ton of data.
- 42:00And you don't need to be because as you mentioned, these
- 42:03checkpoints are huge. And so especially when you're
- 42:06training LLMS that are very, very big, the checkpoints can be
- 42:10in the terabytes. And that's just like if you're
- 42:12doing that every 5 seconds, that's a whole lot of data
- 42:16that's going around. So how did you specifically
- 42:20speed up the checkpoint and make that process better?
- 42:25I think we decided like, so initially the very naive
- 42:31implementation of checkpointing was, you know, so our, our
- 42:35majority of our data is on HDFS, right?
- 42:39And we will the, the mechanism was your checkpoint and the
- 42:44checkpoint goes to SDFS and when you are reading right, you read
- 42:46from there. So the realization specifically,
- 42:50yes, with LLMS like a first of all these checkpoints are big
- 42:54and the training pauses while you are checkpointing, right.
- 43:00So, so there's a pause in the training workload and this gets
- 43:05done. There's a transaction which is
- 43:06happening, right? So the first thing which we did
- 43:09and, and you know, change that architecture as LLMS became more
- 43:13prominent is, hey, let's make it, you know, two phase
- 43:17transaction, right? In essence, we will checkpoint
- 43:21in memory, right? So, and, and from there onwards,
- 43:25any copy of that checkpoint is a sync.
- 43:28So, so two points, right? Like the checkpointing in memory
- 43:32is very fast and the second thing is going on waiting for
- 43:36their checkpoint to be transferred to a remote storage,
- 43:39right. So it's hierarchical
- 43:40checkpointing strategy which we developed and now we are, you
- 43:43know, streaming that checkpoint. We also changed our our back end
- 43:47for checkpointing storage from HDFS to a block based storage.
- 43:51Now we are, you know, investing in, in figuring outgoing and
- 43:55optimizing it even further. So GPU's if you request like
- 44:01when you are ordering GPUs, NVIDIA GPUs, you can request SSD
- 44:05storage within the GPUs, right? And there are technologies which
- 44:12I'm assuming you know some of the companies may have in house,
- 44:16but even in open source, right, where you can then start looking
- 44:19at creating a cache or distributed cache onto the
- 44:23GPUSSD. So maybe 1GPU SSD storage is not
- 44:26enough, right? It's for your needs.
- 44:30Then you can combine the GPU, SSD storage across all your GPUs
- 44:35and then create a distributed cache to store this, right.
- 44:37So that's the second phase we are on in terms of ensuring so
- 44:41that even and and the advantage with that will be in the
- 44:44restore. I think checkpointing will still
- 44:46be fast because it's in memory. It's then we'll go to the
- 44:49GPUSSDS. But when you are reading back,
- 44:52if you don't have to take it outside of the GPU network and
- 44:55bring it back into the GPU network, you save tons while
- 44:58restoring. And when reading, the other
- 45:00thing which we did was like to your .1 of the questions was how
- 45:03often do you checkpoint, right? I mean, that's a big question.
- 45:06And, and we have mostly sort of, you know, left it to the
- 45:11modelers like to take that common sense decision.
- 45:15How soon should you checkpoint right?
- 45:18How much of like for example, for jobs which take like more
- 45:21than a week to train, I think they can have a hey, how much
- 45:25I'm willing to lose? I'm willing to lose 5 hours
- 45:28worth of data or am I willing to lose if if something goes wrong
- 45:31and I have to restart, can I restart from previous days
- 45:34checkpoint? So it's it's a decision making
- 45:36right heuristic in certain use cases, like very deterministic
- 45:41use cases when you're doing incremental training like you
- 45:43need to train every few hours, then it is very clear like hey,
- 45:47it's driven by checkpoints, right.
- 45:49So you have to do a checkpoint, you know, every two hours,
- 45:52etcetera. So depending on the use cases,
- 45:55the other thing which we have done is like disruption
- 45:58readiness. So if in cases of planned
- 46:02maintenance, right, where we know that the nodes are going to
- 46:06go down, right, we trigger a checkpoint, right?
- 46:09So it's a trigger checkpoint. And if the modulus have
- 46:12implemented checkpointing restore, right, that trigger
- 46:15will go in Walker checkpointing restore.
- 46:18So this is not a plan. This is not it's happening
- 46:20before we take that workload of those nodes, right?
- 46:25So essentially this is signal from the underlying infra that
- 46:30hey, these nodes are going away, I'm triggering a notification,
- 46:34this notification and this is all automated right now.
- 46:36We'll go to the the model running model that will be
- 46:39checkpointed moved and then, you know, we will put it back.
- 46:43And then we are invested in things like priority queuing.
- 46:45So if things which are being disrupted, they should be moved
- 46:50to the front of the queue like a Disney front of the line pass
- 46:54and rescheduled like as fast as possible, right?
- 46:56So you need to invest in those mechanisms to make sure that,
- 46:59you know, that goes on smoothly. Yeah.
- 47:03I like the idea of first of all, just getting the checkpoint
- 47:07offloaded as quickly as possible so you can get back to training
- 47:11and then do what you want with the checkpoint, but don't have
- 47:14everything stopped because you're trying to offload the
- 47:18checkpoint. So like make that as quick as
- 47:21possible. I think I've heard that before
- 47:22as as a design pattern and then also make sure that there's no
- 47:27surprises for the model that is training.
- 47:29And then it's like, hey, what happened?
- 47:30I thought I had all these resources and it all went
- 47:33offline. So you make sure that these
- 47:37things are, that they feel kind of standard, but it's like
- 47:42unless you think through it or you have a bad experience, I
- 47:45imagine you probably had an experience where you realized,
- 47:48wow, we should probably do something about that because
- 47:51there was a whole training job that just kind of went to
- 47:55nothing. And we could probably provision
- 47:58or at least put a bit more anticipation in place for that.
- 48:05One thing that I was thinking about is how we've been
- 48:09centering this conversation a lot on the new paradigm of LLMS
- 48:14and agents, but LinkedIn still has a ton of traditional ML use
- 48:20cases, right? So how do you think about
- 48:23bridging the gap or creating a platform that can service both
- 48:29the LLM, the agent use cases and then also the traditional ML use
- 48:36cases? I think it's, it's happening.
- 48:42So certain things are common, right?
- 48:45Like obviously all the investments we are doing in the
- 48:50GPU infrastructure, GPU monitoring, observability,
- 48:53resiliency, and you know, building things like distributed
- 48:59checkpointing, restore automated disruption readiness, then you
- 49:02know, it, it goes, then you start going into like, so one of
- 49:07the things which I did once I came here was, you know, we too,
- 49:12we took a major decision to rehaul our machine learning
- 49:15training pipelines. The pipelines or the
- 49:19orchestration engine was built very much in our solid
- 49:24prescriptive Bay, right? Like it, it was multiple
- 49:28components which we had in our training pipeline.
- 49:31It was based on Tensorflow and, and sort of, you know, it's a
- 49:35similar paradigm which DFX follows Tensorflow extended.
- 49:38You look at from Google and then, you know, every single
- 49:41step is, is sending metadata to the subsystem.
- 49:44So very rigid. Now, I think that realization
- 49:48was very early on here this and, and part of this was also, you
- 49:54know, there is a lot of traditional feature engineering
- 49:56which happens with the traditional Nexus models, right?
- 50:00As opposed to LLMS which don't have the notion of feature
- 50:04engineering, you are feeding, you know, bubbles of text.
- 50:07So, so the feature engineering as a discipline is, is sort of
- 50:12disappearing in, in, in this new world.
- 50:16So, and, and there is a lot of new, you know, rag based
- 50:19architectures which are bringing up right, you're doing more in
- 50:21your time. So the machine learning
- 50:24pipelines engine, we, we rewrote and we redesigned on an open
- 50:29source orchestration engine called flight, because the
- 50:32notion was that experiments will be very heavy right in, in, in
- 50:39this new world. And, and so, you know, there is
- 50:41a lot of quite changing of code. I want to do very short fine
- 50:44tuning jobs or sometimes very long running jobs, but the user
- 50:48experience has to be, you know, very nifty.
- 50:51We launch something called interactive dev where you can,
- 50:55you know how VS Code connect to a remote running job, right?
- 51:00And put debug pointers and trace through what is going wrong
- 51:05because otherwise the previous experience used to be like, you
- 51:07know, you don't know what has gone wrong.
- 51:09So once it has gone wrong, you need to rebuild, recompile, re
- 51:12upload. So this this whole new modern
- 51:16pipeline training architecture which is very much focused on
- 51:19very quick changes, very much experiments.
- 51:21You don't need to re upload. Go directly debug there.
- 51:23See what values are being passed in real time through your debug
- 51:27pointers. Go 1 by 1 and then, you know,
- 51:30once we standardized LLMS on it, we are migrating a lot of our
- 51:35traditional Rexis models onto this new platform as well.
- 51:40And the feedback from users is, is really great.
- 51:43Like they're all liking this, a little bit less rigid, very, you
- 51:48know, experimentation oriented architecture where you know,
- 51:52the, the back and forth things which they need to change,
- 51:55right, to change their model and train their model is, is, and
- 51:59there is a very inbuilt robust versioning mechanism.
- 52:02One thing which was missing in our previous versions of machine
- 52:05learning pipelines was you have a config, you have a model, you
- 52:10have a data set, all these are different version things, right?
- 52:13And you do an experiment, then you have to go and look at every
- 52:17different entity what was correlated with this particular
- 52:20experiment or training run. So in in this new machine
- 52:23learning pipeline, right, there is a very, very robust
- 52:25versioning, right? Like in, in terms of all the
- 52:27entities, your configuration parameters, your pipeline, your
- 52:30model, as well as you know what data version these are all
- 52:33bundled together, right? So every experiment is
- 52:36associated with the right lineage, like, OK, this was all
- 52:38the things which were used to create this particular model,
- 52:41right? So versioning is is very robust
- 52:44because it's sort of built with that assumption.
- 52:46You will experiment a lot, right?
- 52:49So that's one change. I think one thing which
- 52:53potentially and again it's early, early days of the what
- 52:58will change in REXIS with service.
- 53:02You don't want the non LM architecture.
- 53:04I think the LM architecture itself like 1 hypothesis is,
- 53:06hey, the LM architecture itself will take place in REXIS, right.
- 53:10So then OK, whatever you have done at upper layers, right for
- 53:14or you're doing for LMS like vector DBS embedding.
- 53:17So they become the prominent mechanism over all the rag
- 53:20becomes a prominent mechanisms, right.
- 53:22So your traditional feature engineering pipelines, the way
- 53:25you were constructing, they will change, you know, the rear time
- 53:28feature pipelines, not the offline, right.
- 53:29They will change to to accommodate to this new
- 53:31paradigm. Tools like Lang chain, Lang
- 53:33graph, which are very much dominant in in the LLM space.
- 53:37They may become prominent in that space as well.
- 53:41So far we haven't gone that far. I don't think we have looked at
- 53:45hey, can the holder the land chain or land graph is being
- 53:49used right for elements, whether you know prescriptive
- 53:53orchestration graphs when the models are inferencing in real
- 53:56time or non prescriptive agentic graphs which are formed like
- 54:00where the agent decides what part to travels.
- 54:02Can they be used in in the the traditional space where you know
- 54:07there is a lot of feature ingestion happening there
- 54:09featuring engineering pipelines which you have built, But I
- 54:13think the rate is on to see itself.
- 54:18I think the thing which will simplify this can LLMS solve
- 54:21that part of the problem Then you can take the LLM
- 54:23infrastructure as is even at the upper layers and start
- 54:26replicating it for access. Until then, I think it's in fact
- 54:29majority of our workloads are still Rexis workloads are not
- 54:33powered by LLMS, right. And and rightly so, they are
- 54:36very targeted, very effective in what they do.
- 54:40And so it's the first hypothesis has to be at the model layer
- 54:44itself. But at lower layers, we are
- 54:46definitely ensuring that there is standardization across like
- 54:49so it's now the same machine learning pipelines engine which
- 54:52is being used all across LinkedIn, whether it's Forex,
- 54:55Rexis or for LLM workloads, right.
- 54:57So we have standardized all that layer, the topmost layer when
- 55:00you, you know, move towards inferencing and all some of
- 55:03those changes, you know, we'll figure out as we go.
- 55:06Yeah. Yeah.
- 55:07Where, where exactly does it divert?
- 55:09Because it is fascinating to think about how the lower levels
- 55:15you've recognized there, you can share resources or you can have
- 55:19the same type of stuff going on, but at some point it's diverting
- 55:24right now. So where where is that point if
- 55:28you know where it might be case by case basis?
- 55:31I think see first thing is so, so you can look at every single
- 55:35layer, right? Hey, do we need so for example,
- 55:37when you are going to inference, right?
- 55:42Do you, so people have either written their own inferencing
- 55:46engines right in different languages or you have
- 55:48standardized on something like TF serving from open source
- 55:51Tensor Rd. from NVIDIA, which were very heavily optimized for
- 55:55models right in the in the Rexis space, when you look at the LLM
- 56:01space, there is VLLM, there is TGIE, there is SG Lang, right?
- 56:05The inference engine itself, there are custom inference
- 56:08engines being written for LLMS. You just even look at the
- 56:11hardware architecture we didn't talk about, right.
- 56:13There are a lot of custom chips which are coming like you, you
- 56:17see many startups. Grog is another example which
- 56:20are just developing custom hardware.
- 56:24Yeah, yeah, for. Sure.
- 56:25So, so it's, it's if for people who are investing and taking a
- 56:30bet that, hey, it's much simpler for them because they can just
- 56:34standardized on one that the one of the reasons right, where
- 56:37where you may feel, Hey, NVIDIA GPUs could be better for LLMS
- 56:42might be if you write what it says, you know, people who are
- 56:44investing in custom ASICS, custom chips, which are
- 56:47potentially right, better for elements in terms of
- 56:52inferencing, etcetera. Is is the better on the
- 56:54architecture of the transformer architecture is going to replace
- 56:57every single thing, then you can have custom chips.
- 56:59So right, there is some diversion.
- 57:00Some companies are just going with very custom chips, right
- 57:03for elements. So the engine is different.
- 57:06The chips are different, right? Like if you build something
- 57:09general purpose right now, we are we are all, you know,
- 57:12standardized. The hardware layer is is there
- 57:15is no specific thing apart from the kernels work, right, which
- 57:19we have been doing which has been pretty specific.
- 57:22Then you also look at the the the feature processing infra
- 57:27right. So a lot of real time feature
- 57:30processing, feature ingestion in case the kind of tooling which
- 57:36has emerged in the LLM word, it's you are talking, you know,
- 57:40Lanting a land graph vector DBS are prominent, right.
- 57:45It's not the traditional feature processing, right?
- 57:47And and embeddings have become the dominant landscape.
- 57:50And I think we have been on a journey where you know, we are
- 57:53using embeddings all across as well started heavily bought for
- 57:58access as well as LMS. But some of the tooling chain there is
- 58:01different, right, What you will use for something like Beam for
- 58:06example, you know for feature processing in the past now you
- 58:09know a lot of that is happening in the LLM word through the Lang
- 58:13chain and the Lang grass like you know when you are actually
- 58:15orchestrating the data. So those are the divergences
- 58:17which are happening and and are there for the right reasons.
- 58:21I think you know, over a period of time it will reconcile.
- 58:24But yes, that that layer is different when you are handling
- 58:28traditional Rexes versus this, the inferencing engines are
- 58:30different when you are handling there.
- 58:32In some cases people are even taking different chips, hardware
- 58:35chips for these layers. So I think it's bottoms up.
- 58:40We have to push, we have to make sure, you know, continue
- 58:43standardizing on bottom layers, right and go up the stack and
- 58:45see, you know what else you know, as as we become more
- 58:48mature, what else can be standardized and can be the
- 58:51same? If LLM start solving Rexes then
- 58:56potentially maybe you know that itself solves a lot of this
- 58:59problem.