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

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

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

AI 导读

IBM Research 与 UC Berkeley Sky Lab 在开源 K-Search 上新增 MLX 后端和结构化 CUDA-to-MLX 翻译层,让进化式内核搜索把已有 CUDA 内核当作知识库,自动适配成 Apple Silicon 上的 Metal/MLX 内核。

推荐理由

K-Search 的 MLX 后端已开源,在 Apple Silicon 上做本地推理的可以按 README 复现 Attention 与 Mamba 内核的搜索流程。

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