推理与部署AI 资源
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
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
登录后参与讨论,阅读资讯无需注册。
还没有评论,来分享第一个看法。