lightseekorg/tokenspeed

TokenSpeed is a speed-of-light LLM inference engine.

解决的问题

TokenSpeed 旨在提供专为代理工作负载优化的高性能 LLM 推理。它结合了 TensorRT-LLM 的原始速度与 vLLM 的易用性,满足了生产环境中处理 AI 代理特定需求的需求。

运作原理

TokenSpeed 在控制平面与执行平面之间采用了独特的架构分离:

  • 控制平面 (Control Plane):以 C++ 实现的有限状态机。它管理请求生命周期与 KV 缓存所有权,并利用类型系统确保编译时期的资源管理安全。
  • 执行平面 (Execution Plane):以 Python 实现,让研究人员与工程师能快速迭代。
  • 核心系统 (Kernel System):一个模块化、可插拔的系统,具有可移植公开 API 与集中式注册表,包含针对 Blackwell GPU 高度优化的 Multi-head Latent Attention (MLA) 实现。
  • 入口点 (Entrypoint):使用 SMG 集成的 AsyncLLM 来处理低开销的 CPU 端请求。

适用对象

专为部署生产级代理工作负载的工程师与研究人员打造,他们需要在包括 NVIDIA 与 AMD GPU 在内的多种硬件上实现最大推理吞吐量与低延迟。

特色亮点

  • 分离式架构:首个将 C++ 控制平面与 Python 执行平面分离的引擎,兼顾安全性与开发速度。
  • 模块化核心:将核心视为一等公民子系统,并为不同加速器提供可扩展的插件机制。
  • 硬件支持:由非营利组织 LightSeek Foundation 支持,并获得 NVIDIA 与 AMD 的工程支持。
  • 高性能:针对代理任务可达到高吞吐量(例如在 Qwen3.5-397B-A17B 上达到 580 TPS)。

相关

  • 项目
  • 项目
  • 项目
  • 项目
  • Dispatch