Anatomy of a Diffusion Language Model
A breakdown of the modelling choices of three new and popular diffusion language models.
Notes on Building AI Systems
A breakdown of the modelling choices of three new and popular diffusion language models.
Extending SGLang to improve it's offloading capabiltiies, achieving speedups for compute bound workloads.
A follow-on to the economics of speculative decoding, we run the inference lab simulator on MTP & DFlash drafters with real acceptance data, and find out whether adaptively choosing the draft length is worth it.
How a technically superior but economically isolated solution slowly lost ground to a good-enough one built on infrastructure everyone already owns.
How UCCL reimplements the DeepEP kernels for arbitrary hardware by swapping the transport out from under them.
How expert-parallel dispatch and combine kernels work, built up from scratch: the high-throughput shape and the low-latency one.
Two underexplored axes: what MoE routing does to the decode roofline, and how compressed attention takes away the slack that used to make speculated tokens free.
A story of sharp edges, segfaults, and standards
The fastest a memory bound kernel can go is set by the time required to transfer the data to the SMs. How can we do better?
Most conversations in inference have centred on making the experience better for someone waiting. Reducing time-to-first-token. Claude's "fast mode.¹" Groq and Cerebras. The whole technical project of the last few years has assumed that a human, somewhere, is waiting for the response.
Doubleword's batch inference offering keeps costs down by keeping throughput high, something which isn't easily done given the architecture of popular Mixture-of-Expert models. While MoE's sparse expert weights make them quick to train, they also mean that at each layer of every forward each request in a batch typically requires different expert weights to be loaded. This makes inference severely memory-bandwidth bound and cuts throughput relative to dense models. However, by reordering inputs so that similar prompts batch together, we can overlap the experts needed and reduce the number of unique experts loaded per forward. Simply using an embedding model to reorder requests before inference can cut expert loads by approximately 15%, achieving a free throughput gain with no model or kernel changes.
Lossless compression of a target model's KV cache by up to 4×, using a cheaper predictor model to drive an arithmetic coder.