3 ms·
Surpassing vLLM with a Generated Inference Stack
- ntonozzi 7mo agoWhy do they need to run benchmarks to confirm performance? Can't they run an example prompt and verify they get the exact same output token probabilities for all prompts? The fact that they are not doing this makes me suspicious that they are in fact not doing the exact same thing as vLLM. It is also a bit weird that they are not incorporating speculative decoding, that seems like a critical performance optimization, especially for decode heavy workloads.
- lukebechtel 7mo agoYes, speculative decoding will make both us and VLLM faster, but we believe it would be a relatively even bump on both sides, so we didn't include it in this comparison. Worth another test!
- jeeeb 7mo ago> It is also a bit weird that they are not incorporating speculative decoding Wouldn’t speculative decoding decrease overall throughput, but optimise (perceived) responsiveness?
- YetAnotherNick 7mo agoFor compute bound region(high batch size) yes, but for low batch size it could improve the throughput.
- nyrikki 7mo ago> Can't they run an example prompt and verify they get the exact same output token probabilities for all prompts? You don’t even get that with GPUs in general, or really floating point in general. The Art of Computer Programming. Volume 2: Seminumerical Algorithms section 4.2.2 with explain where it loses floating addition associativity property. Apartness relations are another possible lens.
- ntonozzi 7mo agoYeah you can: https://thinkingmachines.ai/blog/defeating-nondeterminism-in-llm-inference/ https://thinkingmachines.ai/blog/defeating-nondeterminism-in....
- nyrikki 7mo ago> However, as the name “batch-invariant” suggests, the technique is currently limited to handling variations related only to the batch dimension, making it robust to continuous batching and other batch-size–related changes, but not to other forms of nondeterminism like changing the TP sizes or GPU types. https://arxiv.org/abs/2506.09501 https://arxiv.org/abs/2506.09501
- rfw300 7mo agoOK... we need way more information than this to validate this claim! I can run Qwen-8B at 1 billion tokens per second if you don't check the model's output quality. No information is given about the source code, correctness, batching, benchmark results, quantization, etc. etc. etc.
- lukebechtel 7mo agoWe validate with MMLU and Hellaswag presently, and are getting this independently verified by a 3rd party. We have considered open-sourcing some of our optimized inference libraries in the future, but have not yet come to a decision on this. Also if you need a rough intuition as to why this is possible: it's because this entire inference stack was built for exactly one model, and thus we can really tune the entire framework accordingly.
- rfw300 7mo agoI've no problem with the intuition. But I would hope for a lot more focus in the marketing materials on proving the (statistical) correctness of the implementation. 15% better inference speed is not worth it to use a completely unknown inference engine not tested across a wide range of generation scenarios.
- lukebechtel 7mo agoThis is a fair critique! We plan to use our system to generate many more inference libraries of this nature, and I'll make it a point to release better, broader correctness measures when we do so.
- LuxBennu 7mo ago[flagged]
- acuozzo 7mo agoLuke: Do you have benchmarks for BF16?
- lukebechtel 7mo agoUnfortunately, not at present; we went for FP8 because we believed it was generally the best tradeoff of quality and speed. Allowed faster iteration as well. We believe our improvements would hold on BF16, but let me check.
- storus 7mo agoDoes it support paged attention like vLLM though? Without that they will run into memory fragmentation quickly.
- lukebechtel 7mo agoYes, great question! The system started without paged attention, and recreated its own paged attention implementation automatically once it realized it was a bottleneck. Pretty cool!
- deleted 7mo ago[deleted]
- cermicelli 7mo ago[flagged]
- 2001zhaozhao 7mo agoEvery example like this makes it obvious that you can now use ML-like optimization approaches on well-specified, very-well-tested software problems with a clear optimization goal. Keep if it improves the objective while maintaining correctness, discard if it doesn't. AI-descent strikes again. Maybe I should learn more about ML to have a better instinct on optimization methods in general, so I can actually build AI optimizers like these.
- lukebechtel 7mo agoThe bitter lesson strikes again, I suppose!
- hoerzu 7mo agoWhat's the jitter what's the std? What about 1:1 output equality? What's the post request latency of this part? What the ftt?
- lukebechtel 7mo agoGood questions! It's clear I need to gather more metrics from our next generated inference library.
- ismailmaj 7mo agoAny place we can find the code?
- lukebechtel 7mo agoUnfortunately it hasn't been open sourced. We're debating how / when to do this right now.
- ismailmaj 7mo agoConfusing, since this is specific to an architecture that no one making money will use (8B is consumer space, not enterprise). The produced code shouldn't hold much interesting IP?