TokenSim: Enabling Hardware and Software Exploration for Large Language Model Inference Systems
摘要
The increasing demand for large language model (LLM) serving has necessitated significant advancements in the optimization and profiling of LLM inference systems. As these models become integral to a wide range of applications, the need for efficient and scalable serving solutions has grown exponentially. This work introduces TokenSim, a comprehensive hardware and software exploration system designed specifically for LLM inference. TokenSim is characterized by its support for extensible system optimizations including scheduling and memory management. Furthermore, TokenSim facilitates various insightful explorations into the performance and optimization of LLM serving systems. The code is available at https://github.com/pku-lemonade/TokenSim .