← 返回列表

gr8:面向 DGX Spark 的 CUDA 推理引擎

kindlingai/gr8 · A from-scratch CUDA inference engine for 1-6 DGX Sparks (GB10): GLM-5.3, GLM-5.3-Flash and DeepSeek-V4.1-Flash

GitHub原文发布 未知首次收录 2026-11-08上架 2026-11-08 07:02

☆ 收藏查看项目 ↗

ELI5:它是什么,解决什么问题

gr8 是一个从零用 CUDA 编写的推理引擎,可在 1 至 6 台 DGX Spark(GB10)上服务 GLM-5.3、GLM-5.3-Flash 和 DeepSeek-V4.1-Flash,并在头节点提供 OpenAI 兼容 API。服务时不使用 vLLM、PyTorch 或 NCCL,通过自身 RDMA 实现张量并行,支持多会话长上下文、前缀复用、KV 卸载与 DeepSeek 图像输入等特性。

核心能力与快速开始

★ 63 · Cuda

能做什么
从零用 CUDA 实现的大模型推理引擎,支持 GLM-5.3、GLM-5.3-Flash 与 DeepSeek-V4.1-Flash,可在 1 至 6 台 DGX Spark(GB10)上以张量并行方式运行,并通过 RoCE v2 RDMA 通信。头节点提供 OpenAI 兼容 API(/v1/chat/completions、/v1/completions、/v1/models、/health)。支持多会话并发与长上下文、前缀复用减少重复 prefill、GLM-5.3 的 KV 缓存卸载到 SSD 或远程 KV 节点,以及 DeepSeek-V4.1-Flash 的可选图像输入。附带 Supervisor 启停、状态、健康检查、smoke 测试与 bench 基准工具。
适用场景
适合在多台 DGX Spark(GB10)集群上私有部署 GLM 或 DeepSeek 系列大模型,并以 OpenAI 兼容接口接入现有应用;适合需要长上下文多轮对话、前缀复用和多并发会话的推理场景;也适合研究 CUDA 原生推理实现与多机 RDMA 张量并行部署的团队。
项目形态
可直接使用的工具
部署方式
提供 Docker 镜像 ghcr.io/kindlingai/gr8 与各机型配方镜像,可用 docker run 或 Docker Compose 启动;文档另提供裸机与 Kubernetes 部署说明。
主要依赖
需 DGX Spark 或其他 GB10 机器(sm_121a,128 GB 统一内存),一台负责一个 rank;多机需要 RoCE v2 双轨网络、TCP 7340/7341 端口,并关闭主动内存整理。自行构建需要 CUDA 13、C++20、CMake、clang/libclang-dev、rdma-core/libibverbs-dev、Rust 1.85+ 与 Python 3 venv。权重需另行下载,体积从约 86 GB 到 510.6 GB 不等。
使用限制
项目为 Beta 发布,作者欢迎反馈问题;模型对机器数量有固定要求(GLM-5.3 为 3 至 6 台,GLM-5.3-Flash 为 1 至 4 台,DeepSeek-V4.1-Flash 为 4 台)。DeepSeek-V4.1-Flash 默认 fast 模式输出与精确模型不完全一致;KV 卸载和会话检查点属实验特性;DeepSeek-V4.1-Flash 没有设备以下的 KV 存储。作者声明 gr8 不是通用引擎。API 成本与商用授权条件未说明。
许可证
MIT
最近推送
2026-11-08 04:59

资料检查:2026-11-08 07:00 · 卡片整理:2026-11-08 07:00
卡片由 AI 根据仓库资料整理,未进行安装或运行验证

这是基于来源内容整理的摘要。原文链接失效后,本页摘要和讨论仍会保留。

讨论 0

登录后参与讨论,阅读资讯无需注册。

还没有评论,来分享第一个看法。