Built independently by an author, for readers. Read the story and support ChapterPal

keyword

inference systems

An inference system is a specialized software and hardware computing framework designed to deploy and execute trained machine learning models to generate predictions or outputs from new data. Unlike model training platforms that focus on updating parameters and calculating gradients, inference systems are engineered for the operational phase of model serving, prioritizing metrics such as low latency, high throughput, and cost-efficient resource utilization. These systems orchestrate execution across computational resources like GPUs, CPUs, and storage devices, coordinating memory management and data flow to handle varying workload demands. To maximize performance under hardware constraints, inference systems often incorporate runtime optimization techniques, including dynamic request batching, model weight and cache compression, precision quantization, tensor scheduling, and distributed offloading strategies.

1 item

FlexGen: High-Throughput Generative Inference of Large Language Models with a Single GPU

FlexGen: High-Throughput Generative Inference of Large Language Models with a Single GPU

Ying Sheng, Lianmin Zheng, Binhang Yuan, Zhuohan Li, Max Ryabinin, Beidi Chen, Percy Liang, Christopher Ré, Ion Stoica, Ce Zhang

OrganizationsCarnegie Mellon UniversityETH ZurichHigher School of EconomicsMetaStanford UniversityUniversity of California BerkeleyYandex

Why you should read this

Presents FlexGen, an offloading engine that combines linear-programming-based tensor scheduling across GPU, CPU, and disk with 4-bit compression to achieve up to 100-fold higher throughput for 175B-parameter model inference on a single commodity GPU.

The high computational and memory requirements of large language model (LLM) inference make it feasible only with multiple high-end accelerators. Motivated by the emerging demand for latency-insensitive tasks with batched processing, this paper initiates the study of high-throughput LLM inference using limited resources, such as a single commodity GPU. We present FlexGen, a high-throughput generation engine for running LLMs with limited GPU memory. FlexGen can be flexibly configured under various hardware resource constraints by aggregating memory and computation from the GPU, CPU, and disk. By solving a linear programming problem, it searches for efficient patterns to store and access tensors. FlexGen further compresses the weights and the attention cache to 4 bits with negligible accuracy loss. These techniques enable FlexGen to have a larger space of batch size choices and thus significantly increase maximum throughput. As a result, when running OPT-175B on a single 16GB GPU, FlexGen achieves significantly higher throughput compared to state-of-the-art offloading systems, reaching a generation throughput of 1 token/s for the first time with an effective batch size of 144. On the HELM benchmark, FlexGen can benchmark a 30B model with a 16GB GPU on 7 representative sub-scenarios in 21 hours. The code is available at https://github.com/FMInference/FlexGen.

Added

2026-09-26