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

K-Search 通过 CUDA 到 MLX 翻译层将内核优化经验迁移到 Apple Silicon

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

AI 导读

研究者基于 UC Berkeley Sky Lab 的 K-Search 演化式内核搜索框架,新增 MLX 后端和结构化 CUDA 到 MLX 翻译层,让现有 CUDA 内核知识自动适配 Apple Silicon。

推荐理由

原文给出结构化 CUDA 到 MLX 翻译层的做法和可复现步骤,读者能了解跨硬件迁移内核优化经验的可行路径。

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