跳到正文
Berkeley AI Research·· 2026-07-29精选AI 评分60

K-Search 扩展 MLX 后端,把 CUDA 内核经验迁移到 Apple Silicon

From CUDA to MLX: How K-Search Brings Decades of Kernel Expertise to Apple Silicon

AI 导读

Berkeley Sky Lab 与 IBM Research 在 K-Search 上新增 MLX 后端和结构化 CUDA-to-MLX 翻译层,让演化式内核搜索以现有 CUDA 内核为知识库生成 Apple Silicon 内核。

借鉴点

可拿走 CUDA 到 MLX 的概念映射表、exp2 技巧和并行 scan 改写,落到本地模型部署;前提是自备 LLM API key 并接受搜索迭代成本。

来源:Berkeley AI Research · bair.berkeley.edu