推理与部署工具
TensorFold:在 Apple Silicon 与 NVIDIA GPU 上提供精确 LLM 解码的 OpenAI 兼容端点
ashhart/TensorFold · Fast, exact LLM decoding on Apple Silicon (MLX) behind an OpenAI-compatible endpoint
ELI5:它是什么,解决什么问题
TensorFold 是一个本地 LLM 推理服务工具,可在 Apple Silicon(MLX/Metal)或 NVIDIA GPU(CUDA)上运行大模型,并通过 OpenAI 兼容的 /v1/chat/completions 端点对外提供服务。用户指定 Hugging Face 上的模型、上下文窗口和采样参数后,它会自动下载模型、加载针对该模型系列编写的内核并启动服务。其核心卖点是“精确”推测解码:通过为每个 token 使用基于种子的 Gumbel 噪声与行级一致的算术,使多行验证与串行解码输出逐字节一致,从而在保持确定性的同时加速解码。仓库提供了模型清单、速度测量、DGX Spark 双机张量并行部署方法以及提示缓存等能力。
核心能力与快速开始
★ 190 · Python
- 能做什么
- 提供 OpenAI 兼容的本地 LLM 服务端点(/v1/chat/completions);支持 Apple Silicon(MLX/Metal)与 NVIDIA GPU(CUDA/PyTorch/Triton)两种后端;从 Hugging Face 按 repo id 下载并加载模型;按模型系列选择专用内核;精确推测解码,宣称多行验证与串行解码逐字节一致;支持上下文窗口、采样参数、思考块与推理预算等配置;DGX Spark 双机张量并行(--tp 2);提示缓存,缓存对话前缀与系统块以加速多轮会话;提供 pull、models、info、update 等 CLI 子命令。
- 适用场景
- 在 Mac 或 NVIDIA GPU 上本地部署 LLM 并以 OpenAI 客户端接入;为编码代理、SDK 或 curl 提供本地模型服务;需要在推测加速下仍保持可复现/确定性输出的场景;在 DGX Spark 双机上切分大模型服务;多轮代理对话中复用缓存前缀以降低预填充开销。
- 项目形态
- 可直接使用的工具
- 部署方式
- 支持 Apple Silicon Mac(macOS,MLX)与 Linux + NVIDIA GPU(CUDA,建议在 NVIDIA PyTorch 容器中运行);DGX Spark 支持双机张量并行。
- 主要依赖
- macOS 需 Apple Silicon 与 Python 3.11 及以上,README 中测试于 MLX 0.31.2(M5 Max)与 MLX 0.32.0(M3 Ultra);不同模型对内存要求不同(如 32 GB、192 GB 以上);CUDA 路径需要 NVIDIA GPU 与含 CUDA toolkit、PyTorch、Triton 的容器。
- 使用限制
- 部分模型对权重量化与内核有条件限制(如 Qwen3.8-27B 的 lane 内核需要 4-bit 分组为 64 的权重与 M5 代 Metal 4 张量单元);NVFP4、GPTQ、AWQ 等无配方权重格式会被拒绝;GLM-5.3-Flash 需两台 Spark,其 DFlash2 草稿模型仅限非商业使用(CC BY-NC-ND 4.0);Flash Next 与 Nemotron 的提示缓存在最后几位可能随已缓存前缀不同而变化。部署条件、硬件与 API 成本等部分信息未说明。
- 许可证
- MIT
- 最近推送
- 2026-09-27 06:47
资料检查:2026-09-27 07:00 · 卡片整理:2026-09-27 07:00
卡片由 AI 根据仓库资料整理,未进行安装或运行验证 · 本次更新未完成,保留上次成功内容
职业应用
程序员
在本地 Apple Silicon 或 NVIDIA GPU 上部署 OpenAI 兼容的 LLM 推理端点,供编码代理、SDK 和 curl 等客户端调用,用于代码生成、工具调用等开发工作。
查看内容依据
Any OpenAI client can then use `http://127.0.0.1:8080/v1`, including coding agents, SDKs and `curl`
由 AI 根据来源材料整理,未进行安装或运行验证
讨论 0
登录后参与讨论,阅读资讯无需注册。
还没有评论,来分享第一个看法。