Exploring the Latency/Throughput & Cost Space for LLM Inference // Timothée Lacroix // CTO Mistral

Join us at our first in-person conference on June 25 all about AI Quality: www.aiqualityconference.com/
// Abstract
Getting the right LLM inference stack means choosing the right model for your task, and running it on the right hardware, with proper inference code. This talk will go through popular inference stacks and set-ups, detailing what makes inference costly. We'll talk about the current generation of open-source models and how to make the best use of them, but we will also touch on features currently missing from the open-source serving stack as well as what the future generations of models will unlock.
// Bio
Timothée Lacroix, aged 31, is Chief Technical Officer in charge of technical issues relating to product efficacy and research. Started as an engineer at Facebook AI Research in 2015 in New York, where he completed his thesis between 2016 and 2019, in collaboration with École des Ponts, on tensor factorization for recommender systems. He continued his career at Meta until 2023 when he co-founded ‪@Mistral-AI‬.
// Sign up for our Newsletter to never miss an event:
mlops.community/join/
// Watch all the conference videos here:
home.mlops.community/home/col...
// Check out the MLOps Community podcast: open.spotify.com/show/7wZygk3...
// Read our blog:
mlops.community/blog
// Join an in-person local meetup near you:
mlops.community/meetups/
// MLOps Swag/Merch:
mlops-community.myshopify.com/
// Follow us on Twitter:
/ mlopscommunity
//Follow us on Linkedin:
/ mlopscommunity

[@evermorecurious91]

This is gold!!!

[@iandanforth]

There seems to be a mistake in the cost estimate at 21:53. It uses the price for the A10 but the throughput of the H100. I believe the actual cost estimate would be $48, not $15.

[@mndflctzn]

This is awesome. Thanks for sharing super useful

[@windmaple]

Great talk!

[@frank96997]

Great talk! is there link to the slides for this talk?

[@Gerald-iz7mv]

hi what benchmark he run to generate the plots? any open source github links?

[@boussouarsari4482]

It's possible that I'm misunderstanding, but given our use of a significantly large key-value cache (2GB multiplied by the batch size), can we still assert that the memory bandwidth is solely influenced by the model's weights?

[@yaxiongzhao6640]

The KV cache's size is directly from the attention layer's size, which in turn is in proportional to model weights' total count So model weights still proportionally determines the kv cache size, thus the statement.

[@janilbolswong1953]

@5:40 why do we need to load the entire model all the time? can't we just load once? If so, we might lower the needs of memory movement, and the intersection would shift left

[@jjh5474]

I guess "memory movement" mean movement from GPU memory(HBM) to GPU computing component. Model parameter stored in GPU memory not in compute component. So for computing model parameter moved from HBM to compute component every forward pass.

[@fraternitas5117]

yes, it needs to be loaded in the gpu all the time. advanced users optimize their applications by sending an equal number of bytes as the memory maximum to optimize the utilizations of all memory in the clock cycle.

[@aneeinaec]

Is that Ryan Gosling ❤

[@eduardoalvarez7152]

The math around 6:50 for A100 batch size isn't working out. It would be great if the values used to calculate the 400 batch size were provided. Based on the equations provided for compute time and model load time, the point of intersection is Flops/(2*MemoryBand) NOT the (2*FLOPS)/MemoryBand which is in the video.

[@TheAIEpiphany]

I believe it was just a piece of napkin math: in reality he didn't count in KV cache at all in the P / mem bandwidth line which is a function of sequence length. That seems like the biggest approximation error I see here? For the second line he discounted attention FLOPs and used just MLP FLOPs (the error of this approximation increases as the sequence grows, depends on the model size you're using e.g. for 7B model with a big sequence length, that term might actually be important). Additionally the peak flops is a function of the data type and the operation you're executing, he's assuming bf16/fp16 which is what Mistral 7B is using, that gives you ~312 TFLOPs/s for A100. All in all this is useful if you understand exactly the assumptions he's making.

[@Venkat2811]

@@TheAIEpiphany Yes, I was looking for KV cache as well. Your explanation makes sense.

[@MLOps]

Join us at our first in-person conference on June 25 all about AI Quality: www.aiqualityconference.com/

[@AbdulK-kr2jv]

What a horrible unethical response on the ethics of training data