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

从 CUDA 到 MLX:K-Search 如何把内核优化经验带到 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。

推荐理由

K-Search 把 CUDA 内核经验迁移到 Apple Silicon,读者可看到跨平台内核优化的具体方法与实测数字。

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