Berkeley AI Research·· 2026-07-29精选AI 评分65
K-Search 如何将数十年 CUDA 内核经验迁移到 Apple Silicon 的 MLX
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 到 MLX 翻译层，在 Apple Silicon 上达到接近专家水平的内核性能。
推荐理由
原文展示 CUDA 优化知识如何转成 MLX 架构策略，并给出注意力与 Mamba 预填充的实测对比，便于理解跨硬件内核优化如何迁移。
来源：Berkeley AI Research · bair.berkeley.edu