EAGLE-4 的推理代码,通过投机解码为 SGLang、vLLM 和 TokenSpeed 上的大模型推理提速。
snowflakedb/ArcticInference
vLLM 插件,集成 Snowflake 的推理优化(Shift Parallelism、投机解码、SwiftKV),用于加速并降低 LLM 与嵌入模型的服务成本。
- Star
- 491
- Fork
- 69
- 未关闭 Issue
- 66
- 最近提交
- 上个月
这是什么?
Arctic Inference 是面向 vLLM(主流 LLM 推理引擎)的开源插件。安装后会自动接管 vLLM,使现有的 vLLM API 和命令行工具无需修改代码即可使用其优化,包括 Arctic Ulysses 序列并行、Shift Parallelism、Arctic Speculator 与 Suffix Decoding 等投机解码方案、SwiftKV 等模型层优化,以及面向嵌入模型和推理类任务的专门路径。
项目覆盖 LLM 文本生成和嵌入模型推理两类负载。仓库提供了用 vLLM 启动服务和使用 LLM 类进行离线推理的示例,只需设置环境变量 ARCTIC_INFERENCE_ENABLED=1 即可启用。通过 pip 安装并指定 vllm 依赖即可使用。
核心功能
- 01提供 Shift Parallelism 与 Ulysses 序列并行,用于 LLM 服务
- 02通过 Arctic Speculator 和 Suffix Decoding 实现投机解码
- 03支持 SwiftKV 模型层优化
- 04针对 vLLM 优化的嵌入模型推理
- 05兼容现有 vLLM API 与命令行,通过环境变量启用
适用场景
- 在 vLLM 上服务大模型,同时降低延迟、提高吞吐
- 在 GPU 上运行大批量嵌入向量生成任务
- 为长上下文或推理密集型负载降低推理成本
快速上手
- 1
安装带 vLLM 额外依赖的包
pip install arctic-inference[vllm] - 2
启用 Arctic Inference 启动模型服务
ARCTIC_INFERENCE_ENABLED=1 vllm serve Snowflake/Llama-3.1-SwiftKV-8B-Instruct --quantization "fp8" --tensor-parallel-size 1 --ulysses-sequence-parallel-size 2 --enable-shift-parallel --speculative-config '{"method": "arctic", "model":"Snowflake/Arctic-LSTM-Speculator-Llama-3.1-8B-Instruct", "num_speculative_tokens": 3, "enable_suffix_decoding": true, "disable_by_batch_size": 64}'
使用前须知
- 针对 vLLM 优化,需要配备 vLLM 的基于 GPU 的推理环境。
- 推测解码和 Shift Parallelism 的设置针对特定 Snowflake 模型调优,需要自行配置。
收录于专题
标签
常见问题
- ArcticInference 是什么?
- vLLM 插件,集成 Snowflake 的推理优化(Shift Parallelism、投机解码、SwiftKV),用于加速并降低 LLM 与嵌入模型的服务成本。 Arctic Inference 是面向 vLLM(主流 LLM 推理引擎)的开源插件。安装后会自动接管 vLLM,使现有的 vLLM API 和命令行工具无需修改代码即可使用其优化,包括 Arctic Ulysses 序列并行、Shift Parallelism、Arctic Speculator 与 Suffix Decoding 等投机解码方案、SwiftKV 等模型层优化,以及面向嵌入模型和推理类任务的专门路径。
- ArcticInference 可以免费使用吗?用的什么许可证?
- ArcticInference 以 Apache-2.0 许可证开源。用于商业产品前,请先确认许可证条款。
- ArcticInference 还在维护吗?
- 半年内有更新。ArcticInference 最近一次提交是在 2026年9月23日。
- ArcticInference 适合用来做什么?
- 在 vLLM 上服务大模型,同时降低延迟、提高吞吐 在 GPU 上运行大批量嵌入向量生成任务 为长上下文或推理密集型负载降低推理成本
- ArcticInference 怎么安装和运行?
- 安装带 vLLM 额外依赖的包: pip install arctic-inference[vllm] 启用 Arctic Inference 启动模型服务: ARCTIC_INFERENCE_ENABLED=1 vllm serve Snowflake/Llama-3.1-SwiftKV-8B-Instruct --quantization "fp8" --tensor-parallel-size 1 --ulysses-sequence-parallel-size 2 --enable-shift-parallel --speculative-config '{"method": "arctic", "model":"Snowflake/Arctic-LSTM-Speculator-Llama-3.1-8B-Instruct", "num_speculative_tokens": 3, "enable_suffix_decoding": true, "disable_by_batch_size": 64}'
- 在哪里获取 ArcticInference?
- 源代码和安装说明在 GitHub 上:https://github.com/snowflakedb/ArcticInference
概要由 AI 根据项目 README 生成;GitHub 数据每日更新。
相关项目
用于训练推测解码草稿模型、可直接部署到 vLLM 推理引擎的 Python 库。
DeepSeek 开源的离线推理框架,用于在 NVIDIA SM100 和华为昇腾 950 硬件上运行 DeepSeek-V4.1-Flash。
面向消费级 GPU 与异构机器的 C++ 大模型推理服务,结合投机解码、投机预填充与定制内核。
面向数据中心规模的分布式推理服务框架,可协调 vLLM、SGLang 和 TensorRT-LLM 跨多 GPU、多节点运行。
面向 Apple silicon Mac 的本地大模型推理引擎,兼容 OpenAI 与 Anthropic 接口,可直接运行编码智能体。