JAX 可跑 Apple GPU：jax-graft 用 Opus 5.5 写成开源后端
twiecki · x · 2026-10-12
天文学家 dfm 一直希望在 Apple GPU 上运行 JAX，于是用 Claude Opus 5.5 构建了适配新版 JAX 的自定义后端 jax-graft，现已在 GitHub 开源（Apache-2.0）。
实现思路：把 Metal 当作一个 XLA:GPU 平台（名为 "mtl"），与 CUDA、ROCm 并列。XLA 的 GPU 编译器负责程序融合与内核生成，插件再将生成的内核翻译成 Metal Shading Language。项目不与 Apple 官方关联，作者承认还有不少粗糙之处，但「基本能跑」。
这对 Mac 本地 AI/科学计算生态是个有意思的信号：AI 编程助手已经能独立产出涉及编译器栈的复杂系统项目。
「编程与Agent」频道最新
- 社区众包+AI agent 一天狂提 69%，整数乘法 κ 界破 10⁻³ — CatAstro_Piyush · 2026-10-12
- AI 智能体端到端部署微型工厂：2 万欧元两年回本 — ihorbeaver · 2026-10-12
- 独立开发者用 Claude 打造开源套件，单挑 Adobe 与微软 Office — mark_k · 2026-10-12
- 开源 xtool 登场：在 Linux/Windows 上构建并部署 iOS 应用 — sull · 2026-10-12
- Teknium：线框图与原型设计 skill 正式并入 Hermes Agent — marktenenholtz · 2026-10-12
- 网友汇总 870 条回复整理出 263 个 Grok Bot 真实用例 — thisiskp_ · 2026-10-12