Nyla Worker: evaluation

Transcript context

…When I started was first scaling up training and making training more efficient. And that of course has evolved significantly over time. There is a lot on training. But what I discovered is that if these things are truly useful, you should be obsessing about inference. And then I went to eBay, uh, where I was in their hardware team, but I was doing software optimizations for the hardware team, such that the research that had been done for the AI research team was actually running efficiently on the hardware. And there, I started leveraging optimization, uh, frameworks such as TensorRT to optimize our models like ResNet 50. So the way that the, uh, AI research team at eBay had implemented image search was some kind of computer vision model, and then we would retrieve an embedding from a certain layer of this ResNet 50 model, and then do some kind of distance with the other images. And it was very advanced for the time, and what I had to do was to make it more efficient. So the way that it went to production actually was A single image before the ResNet 50, meaning batch one, and it was running with a certain latency. But there were product requirements, right? And this is where inference becomes very interesting because it's not about making it the fastest, it's about meeting the human perceived latency. Right? And in this case, what we realized is that for this particular case was seven milliseconds For the particular inference of the model. And then obviously wrapped up in the whole service probably was going to be under 50 or 100 milliseconds, which is unperceptible to humans. So in that, my objective was to get the more bang out of back of the hardware. And we were evaluating different hardwares, but my particular focus was on a V100 and we optimized it with TensorRT. And TensorRT has, uh, does a lot in the backend. So for example, it fuses kernels, it quantizes the model, it reduces that precision. Of course, now everyone talks about quantization, but then it was like FP32 to FP16. Intel was still like very, very early. And even then, we went from having a service in production with one image to four images in seven milliseconds. And we got that running quite effectively. So, since then, however, what we've seen with that same model, right? At that time, it was TensorRT. Resnet 50 2018. Uh, four images for seven milliseconds. If you do the rough calculation, that is a throughput of about 571. And if you look at the efficiencies that have been gained over the past couple of years, and this is running on a V 100, which is not optimized, you can check the numbers from last year from ML PERF and see that now it's 88,000. iencies that have been gained over the past couple of years, and this is running on a V 100, which is not optimized, you can check the numbers from last year from ML PERF and see that now it's 88,000. Images or samples per second. They use samples. And obviously this is not necessarily apples to apples comparison because you need to check at the fine print as to how they are running this. They are not optimizing for latency. Um, so they are optimizing for 2. 0 first, but even then, like that number is like, It's striking, right? And there are other things that I learned through my time at Nvidia. So, and I can dive more into that, but if you have anything to add there. Yeah, no, that's great. And I think especially the hardware piece is really important. Like, uh, back when you were at eBay, you mentioned the V100 was kind of state of the art. The v100 is about 130 teraflops of kind of like compute the gb200 at fp4 is like 20, 000 teraflops so the hardware alone today got much more powerful and I would love to maybe hear from you how at the time you were thinking about optimizing for the hardware today versus how much of an insight you had into the hardware that was coming especially working at NVIDIA and maybe people have the same discussion today it's like you know Should we optimize for the hardware of today or like for the hardware of tomorrow, because we need the results today, you know, as a business, but sometimes maybe we waste some time. So curious to hear your thoughts.…

Stored transcript either side of the excerpt. The highlighted words are the published quote; the surrounding text is unedited source, never generated.