AIBrix优化大模型推理基础设施,降本增效并提升稳定性。
AIBrix: Towards Scalable, Cost-Effective Large Language Model Inference Infrastructure
- 分层协同设计,适配vLLM等推理引擎,支持动态LoRA调度。
- 分布式KV缓存使吞吐提升50%,延迟降低70%。
- 适合需要高性价比、大规模部署的AI服务团队。
我们提出AIBrix,一个云原生开源框架,旨在优化和简化云端大规模大语言模型的部署。与传统云原生架构不同,AIBrix采用协同设计思想,使每一层基础设施均针对vLLM等推理引擎量身打造。其关键创新包括高密度LoRA管理实现动态适配器调度、面向LLM的自动扩缩容机制,以及基于前缀和负载感知的路由策略。为提升效率,AIBrix引入分布式KV缓存,跨节点增强令牌复用,使吞吐量提升50%,推理延迟降低70%。该框架还支持统一AI运行时,简化模型管理并保持对多种引擎的兼容性。在多节点场景下,采用混合编排策略:利用Kubernetes进行粗粒度调度,通过Ray实现细粒度执行,兼顾效率与灵活性。此外,基于SLO的GPU优化器可动态调整资源分配,在异构服务中最大化成本效益并保障服务质量。系统可靠性方面,AIBrix集成AI加速器诊断工具,支持自动故障检测与模拟测试,提升容错能力。项目已开源,地址为https://github.com/vllm-project/aibrix。
原文摘要 · Abstract (English)
We introduce AIBrix, a cloud-native, open-source framework designed to optimize and simplify large-scale LLM deployment in cloud environments. Unlike traditional cloud-native stacks, AIBrix follows a co-design philosophy, ensuring every layer of the infrastructure is purpose-built for seamless integration with inference engines like vLLM. AIBrix introduces several key innovations to reduce inference costs and enhance performance including high-density LoRA management for dynamic adapter scheduling, LLM-specific autoscalers, and prefix-aware, load-aware routing. To further improve efficiency, AIBrix incorporates a distributed KV cache, boosting token reuse across nodes, leading to a 50% increase in throughput and a 70% reduction in inference latency. AIBrix also supports unified AI runtime which streamlines model management while maintaining vendor-agnostic engine compatibility. For large-scale multi-node inference, AIBrix employs hybrid orchestration -- leveraging Kubernetes for coarse-grained scheduling and Ray for fine-grained execution -- to balance efficiency and flexibility. Additionally, an SLO-driven GPU optimizer dynamically adjusts resource allocations, optimizing heterogeneous serving to maximize cost efficiency while maintaining service guarantees. Finally, AIBrix enhances system reliability with AI accelerator diagnostic tools, enabling automated failure detection and mock-up testing to improve fault resilience. AIBrix is available at https://github.com/vllm-project/aibrix.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。