Video yükleniyor...

Video Yüklenemedi

Ana Sayfaya Dön

While exploring diffusion language models (DLMs), we kept asking ourselves: given speculative decoding, a lossless acceleration technique, why do we still need DLMs? 🤔 Our latest work, ORL, tries to answer this question: AR optimizes intelligence 👆, speculative decoding optimizes speed 👉, and DLMs optimize the Pareto frontier of...

20,814 görüntüleme • 4 gün önce •via X (Twitter)

14 Yorum

Chenfeng_X profil fotoğrafı
Chenfeng_X4 gün önce

Three highlights for DLM communities 😄: 1. ORL points out that on-policy residual learning is the key to pushing a DLM's tokens per forward (TPF) to the limit. We release an RL training framework that implements exact on-policy residual capture, directly overcoming the state-staleness problem common to residual DLMs and recent popular loop transformers. Check out the GitHub: 2. ORL introduces an inference engine for DLMs. It is customized for low-latency, single-request DLM workloads. Check out the Github: 3. Block attention in DLMs has never matched FlashAttention-like speed in AR, so we redesigned the kernel for block-causal attention. Check out how we design the kernels: @Yifan_YYU, @jianyuqing1, @jisenli_ai , @yuezhouhu

Chenfeng_X profil fotoğrafı
Chenfeng_X4 gün önce

ORL enables more parallelism at comparable quality: up to 6.2× TPF on QwenDLM (accuracy up on all four benchmarks), up to 1.5× TPF on LLaDA-2.1-mini, and up to 1.2× TPF on DiffusionGemma. With our engine and NVFP4, a single request decodes at 5,046 tok/s on one B200, 7,708 tok/s on 8× B200, and 2,788 tok/s on a consumer RTX 5090.

Chenfeng_X profil fotoğrafı
Chenfeng_X4 gün önce

Insight 1: residual learning is inherently on-policy. A model's residuals decide which tokens get committed, and so which states it sees next. Offline residuals: TPF barely moves. Stale trajectories: accuracy drops. Fresh on-policy rollouts: both improve.

Chenfeng_X profil fotoğrafı
Chenfeng_X4 gün önce

Insight 2: Lower cross-entropy loss ≠ refinement speed. Under standard training, entropy stays flat across denoising steps. ORL adds adjacent entropy contraction: each step must cut the uncertainty the previous one left. A block finishes in 22 steps instead of 41.

Chenfeng_X profil fotoğrafı
Chenfeng_X4 gün önce

Insight 3: exact on-policy residual learning is a new systems workload. Considering per-step states, irregular trajectories, and heavy memory, we co-design async capture, memory-adaptive transport, packed compute-balanced training, and a staircase attention kernel: up to 5.2× faster training.

Chenfeng_X profil fotoğrafı
Chenfeng_X4 gün önce

Result 1: faster than speculative decoding, lossless or lossy. Single request on one H100, avg GSM8K/HumanEval: AR (Qwen3-4B): 240 tok/s + EAGLE3: 353 strict / 405 relaxed + DFlash: 573 strict / 746 relaxed QwenDLM + ORL: 1,298 tok/s That's 1.74× faster than DFlash at matched accuracy. Even the lossiest DFlash setting we measured tops out at 827 tok/s, at 80% accuracy. Same on Gemma-4-26B: DiffusionGemma + ORL runs at ~1,400 tok/s, vs at most 663 tok/s for any Gemma speculative setting.

Chenfeng_X profil fotoğrafı
Chenfeng_X4 gün önce

Result 2: ORL moves the whole quality–throughput frontier, not just one operating point. Sweeping the decoding threshold, ORL's curve sits above and to the right of the base model's: 5.6× higher throughput on QwenDLM and 2.95× on LLaDA-2.1-mini at comparable accuracy. The gain grows with block size: on HumanEval, ORL lifts LLaDA-2.1-mini's TPF from 11.4 to 15.0 at block size 32, and from 9.2 to 19.2 at block size 128, with accuracy up at every size.

Chenfeng_X profil fotoğrafı
Chenfeng_X4 gün önce

Result 3: little data, little compute. ORL learns from the model's own rollouts. On LLaDA-2.1-mini, it needs only 4.2M self-generated tokens and 36 H100-hours of post-training, about 10× less data than the ~40M tokens used by d3LLM-LLaDA and dParallel.

Chenfeng_X profil fotoğrafı
Chenfeng_X4 gün önce

Result 4: an engine built for a single request. Serving engines are built around batching. At batch size 1, a DLM's forward pass is short, so kernel launches and scheduling eat a large share of it. Our engine runs each block's denoising loop on the GPU as one CUDA graph: 1.22–1.56× faster than SGLang and vLLM on the same checkpoints, without changing the decoding rule (4,816 of 4,820 QwenDLM answers are identical to SGLang's).

Chenfeng_X profil fotoğrafı
Chenfeng_X4 gün önce

Result 5: fast on real hardware, even consumer GPUs. ORL cuts the number of forward passes, and NVFP4 makes each pass cheaper, so the two compose. On one B200, NVFP4 is 2.3–2.9× faster than BF16 on one H100, with small accuracy changes. DiffusionGemma + ORL in NVFP4 is also 1.14–1.30× faster than NVIDIA's own NVFP4 DiffusionGemma at comparable accuracy. And the NVFP4 ORL models fit on a single 32 GB RTX 5090, matching or beating BF16 on an H100.

Chenfeng_X profil fotoğrafı
Chenfeng_X4 gün önce

Code, models, and the engine are open. 🌐 Project + live demos: 📄 Paper: 💻 ⚡ Joint work with an amazing team: @Yifan_YYU , @jianyuqing1, @jisenli_ai , @yuezhouhu , Tianyi Zhang, @XiaoxiaWShirley , @RidgerZhu , @_junxiong_wang , @ZhongzhuZhou , Fan Lai, and @james_y_zou , across @togethercompute, UIUC, UC Berkeley, UC Santa Cruz, Stanford, and UT Austin.

Sharon Lee profil fotoğrafı
Sharon Lee4 gün önce

Chenfeng !

Anthony profil fotoğrafı
Anthony4 gün önce

Amazing 🤩🐐🐐

shaped profil fotoğrafı
shaped4 gün önce

would love to see this tested on longer agent trajectories where decoding costs really add up

Benzer Videolar

Researchers found a way to make LLMs 8.5x faster! (without compromising accuracy) Speculative decoding is quite an effective way to address the single-token bottleneck in traditional LLM inference. A small "draft" model first generates the next several tokens, then the large model verifies all of them at once in a single forward pass. If a token at any position is wrong, you keep everything before it and restart from there. This never does worse than normal decoding. But current drafters in Speculative decoding still guess one token at a time. That makes the drafting step itself a bottleneck, capping real-world speedups at 2-3x. DFlash is a new technique that swaps the autoregressive drafter with a lightweight block diffusion model that guesses all tokens in one parallel shot. Drafting cost stays flat no matter how many tokens you speculate. On top of that, the drafter is conditioned on hidden features pulled from multiple layers of the target model and injected into every draft layer, so it makes significantly better guesses than a drafter working from scratch. In the side-by-side demo below, vanilla decoding runs at 48.5 tokens/sec. DFlash hits 415 tokens/sec on the same model, with zero quality loss. It's already integrated with vLLM, SGLang, and Transformers, with draft models on HuggingFace for several models like Qwen3, Qwen3.5, Llama 3.1, Kimi-K2.5, gpt-oss, and many more. I have shared the GitHub repo in the replies! KV caching is another must-know technique to boost LLM inference. I recently wrote an article about it. Read it below. 👉 Over to you: What use case are you working on that can benefit from this new technique?

Avi Chawla

158,352 görüntüleme • 5 ay önce

This inference engine runs LLMs 4x faster than llama.cpp: (Qwen3.5-9B at 92 tokens/s on 16GB Mac) I ran Qwen3.5 9B on a base M5 MacBook Pro using three local inference engines, one at a time. → llama.cpp reached 22.0 tokens/s → MLX reached 25.1 tokens/s → Uzu reached 92.1 tokens/s The video shows all three runs side by side. The first two results do not mean llama.cpp or MLX are poorly optimized. Both are approaching a physical limit of ordinary local decoding. Generating each new token normally requires another pass through the model. That pass streams roughly 5.2 GB of weights through memory. A base M5 provides about 153 GB/s of memory bandwidth. At one token per pass, the theoretical ceiling is roughly 153 ÷ 5.2, or 29 tokens/s, however fast the GPU computes. Uzu gets past that ceiling by optimizing speculative decoding for Apple Silicon. Speculative decoding uses a smaller drafting system to propose several future tokens. The full 9B model then checks those proposals together instead of generating every token through a separate pass. Uzu pushes this further in two ways. → It drafts up to 16 positions and uses Weaver to turn independent guesses into coherent sequences. → It verifies several possible sequences as a tree, then commits state only for the path the full model accepts. The 9B model remains in control of the final output. Uzu simply gives it more useful work to approve in each pass. In my run, every pass of the 9B model produced 7.5 output tokens on average. Ordinary decoding produces exactly 1. That is how Uzu reached 92.1 tokens/s on hardware whose one-token-per-pass ceiling is roughly 29. The rest of the engine is designed around this path too. Uzu ships its own quantized checkpoints, and on M5 chips its Metal kernels run verification through the GPU’s accelerated int8 matrix path. It is an open-source inference engine from Mirai, with bindings for Rust, Swift, Python, and TypeScript. The repository is here: (don't forget to star 🌟) I wrote the full breakdown of why local inference is memory-bound and how Uzu gets past that limit, with reproducible results and code. The article is quoted below.

Akshay 🚀

81,359 görüntüleme • 10 gün önce

Baseten Head of AI Model Training Charlie O'Neill says the future is many specialized LLMs dedicated to specific tasks, with bigger labs deployed on the frontiers of areas like science and math: "People are thinking about intelligence capabilities in the wrong way. People are thinking about intelligence relativistically. They say, 'OK, the open-source gap is like 6 months behind closed-source, and GLM 5.3 is as good as Opus 4.8,' or whatever." "The best way to think about what models can do for you, and for the world, is in an absolute sense." "So for any given task that you want to do with an LLM, there's some intelligence threshold where below that you can't do the task, and above that you have very diminishing returns to more intelligence on the task." "So when you think about it that way, the game of LLMs over the last 5 years has been, 'OK, we have these things we want to do with them. Closed source hits it first... but open-source can eventually do that task. And then for many reasons, once you have the base level of intelligence required to do it, you probably do want to swap to open-source." "It's not really about the [frontier lab] God model being better. Like, if I'm filing a tax return, there is a limit to how much intelligence I need to do that particular thing." "So I think the world is going to look like — frontier closed-source labs are going to continue to push the frontier. You do want to use the most intelligent model. You have very inelastic demand for intelligence when you're doing frontier science or frontier math." "But for a lot of the economically valuable things, it looks a lot like, 'I'm a Cursor, or I'm one of these big companies who are realizing I can't just be a wrapper anymore. I've been through the life cycle of building a product that people love. And I should be using that information to make my model better at the things that I care about, and not at anything else.'"

TBPN

50,531 görüntüleme • 1 ay önce

There is no best model. There's a lot of noise about models right now. Who is training them, who owns them, where legal intelligence should live. One question actually matters: what produces the best outcome for the legal task in front of you? That's how we decide things at Legora. We optimize for the end-to-end outcome on a legal task. The model is one layer of that system, not the system. Models are uneven and the frontier changes almost weekly. One model plans a long job well, another runs deep analysis across thousands of documents. Some have to be told exactly what to do, and some are fine with a vague brief. They all break in different ways. So our lawyers write evals and we test them with the Legora BAR, our benchmark for agentic reasoning. Every model takes every test, and the model that wins gets the work. We post-train when we know it buys our customers better performance on a specialized task. Training is a tool we reach for when it helps, nothing more than that. The intelligence that compounds sits in the orchestration layer. Precedents, review standards, client requirements. That knowledge has to stay editable, auditable and portable. In our system, a changed review standard is an edit that takes effect the same day, with no new model training required. No lawyer should have to worry about which model did the work, any more than they think about which chip is in their laptop. They should only care about the quality of the work. That's what we are focused on. If you want the engineering version of this argument rather than the CEO version, our CPO, Bryan Tsao, and CTO, Jacob Lauritzen, take it apart in the video below.

Max Junestrand

49,595 görüntüleme • 1 ay önce

I had to test it myself to believe this unreal inference speed. 3,000 tokens/s for 1 user on standard datacenter GPUs. They leveraged a hidden efficiency gap in how GPUs generate tokens. Kog just achieved 3,000 tokens/s on 8× AMD MI300X GPUs and 2,100 on 8× NVIDIA H200 (FP16, no speculative decoding). Their tech preview is on a 2B model, and they show how their techniques will scale to large frontier MoE models at similar speeds. That's a huge number because normal low-batch GPU decoding for 2B to 8B models is usually closer to 100 to 300 tokens/s per request, so Kog is claiming something like a 10X to 30X jump in the speed one user actually feels. Their trick: they are getting the speed by treating LLM decoding as a memory streaming problem, not mainly a math problem. For 1 user at batch size 1, the GPU is not doing big, efficient matrix-matrix work like in training or large-batch serving; it is repeatedly pulling the model’s active weights from high-bandwidth memory for each new token, so speed depends on how smoothly those weights keep flowing. Normal inference stacks keep breaking that flow. They run many separate GPU programs for different parts of the model, move intermediate results through memory, wait at synchronization points, talk back to the CPU for scheduling or sampling, and then repeat this token after token. Kog’s answer is to co-design 3 things that are usually tuned separately: the runtime, the low-level GPU code, and the model architecture. The biggest engineering move is the monokernel, where the whole decode pass runs as 1 persistent GPU-resident program, including sampling, so the system does not keep stopping for kernel launches, CPU scheduling, and intermediate memory round trips. They also rebuilt synchronization, because their own measurements say grid sync was eating around 35% of token-generation time; instead of making every compute unit wait at a broad barrier, each unit waits only for the exact data it needs. On AMD MI300X, they also map memory access around the chiplet layout, because memory latency changes depending on which die makes the request. Then their Laneformer model uses Delayed Tensor Parallelism, which lets cross-GPU communication happen in the background instead of blocking every layer.

Rohan Paul

13,282 görüntüleme • 4 ay önce

Today, we’re excited to announce our $50M Series B, led by Greenfield Partners (formerly TPG Capital), with participation from Lightspeed and Notable Capital. 🚀 At PatronusAI, we develop simulations and evals to train and improve AI. The first phase of AI was built on static benchmarks, but that era is over now. As agents are used to solve longer and longer tasks, they need to practice in dynamic, living worlds to get better. Simulations are the critical infrastructure powering this next phase. As a company, we’re behind the most influential research and products in AI evaluation, like FinanceBench, Lynx, and Percival. And things have moved at the speed of light since. ⚡ We partner with the world's leading frontier AI labs and enterprises, and our revenue has grown more than 15x over the past year. Additionally, today, we’re introducing a preview of the first Digital World Model for AI agent training and simulation: Patronus-DWM. Digital World Models are language diffusion world models that predict realistic environment behaviors and steer agent actions across digital workflows. Just as physical world models predict how objects move through space, we’re developing the equivalent for the digital world: predicting how agents act in digital workflows, then using that to scale the creation of high-quality training data for LLMs. Digital World Models help us push the frontier of ultra long horizon workflows, and unlock a new class of self-improving RL environments. This is our scalable approach to simulating all of the world’s intelligence. The round was also joined by Datadog, Inc., Samsung Ventures, Gokul Rajaram, Factorial Capital, and a large cohort of amazing AI leaders and researchers across Anthropic, OpenAI, Google DeepMind, NVIDIA, Recursive, and more. ✨ It has been the ride of a lifetime. But we’re just getting started. The best is yet to come. "Do not go gentle into that good night, Rage, rage against the dying of the light" - Dylan Thomas (1954)

Anand Kannappan

45,854 görüntüleme • 3 ay önce

Today, we’re excited to announce our $50M Series B, led by Greenfield Partners, with participation from Lightspeed and Notable Capital. 🚀 At Patronus AI, we develop simulations and evals to train and improve AI. The first phase of AI was built on static benchmarks, but that era is over. As agents are used to solve longer and longer tasks, they need to practice in dynamic, living worlds to get better. Simulations are the critical infrastructure powering this next phase. As a company, we’re behind the most influential research and products in AI evaluation, like FinanceBench, Lynx, and Percival. And things have moved at the speed of light since.⚡ We partner with the world's leading frontier AI labs and enterprises, and our revenue has grown more than 15x over the past year. Additionally, today, we’re introducing a preview of the first Digital World Model for AI agent training and simulation: Patronus-DWM. Digital World Models are language diffusion world models that predict realistic environment behaviors and steer agent actions across digital workflows. Just as physical world models predict how objects move through space, we’re developing the equivalent for the digital world: predicting how agents act in digital workflows, then using that to scale the creation of high-quality training data for LLMs. Digital World Models help us push the frontier of ultra long horizon workflows, and unlock a new class of self-improving RL environments. This is our scalable approach to simulating all of the world’s intelligence. The round was also joined by Datadog, Inc., Samsung Ventures, Gokul Rajaram, Factorial Capital, and a large cohort of amazing AI leaders across Anthropic, OpenAI, Google DeepMind, NVIDIA, Recursive, and more.✨ It has been the ride of a lifetime. But we’re just getting started. The best is yet to come. "Do not go gentle into that good night, Rage, rage against the dying of the light" - Dylan Thomas (1954)

PatronusAI

95,919 görüntüleme • 3 ay önce