Berkeley AI Research·· 2026-07-29精选AI 评分64
K-Search 如何将 CUDA 内核经验迁移到 Apple Silicon
From CUDA to MLX: How K-Search Brings Decades of Kernel Expertise to Apple Silicon
AI 导读
IBM Research 将结构化 CUDA-to-MLX 翻译层接入 K-Search,使其能把 CUDA 内核知识迁移为 Apple Silicon 上的 MLX 内核。
推荐理由
文章展示了 CUDA 优化经验如何通过翻译层迁移到 MLX,并用 Attention 与 Mamba 的实测结果呈现这一路径的效果。
来源:Berkeley AI Research · bair.berkeley.edu