跳到正文
原文
BAIR:Berkeley AI Research Blog·· 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 导读

IBM Research 基于 UC Berkeley Sky Lab 的 K-Search 框架扩展出 MLX 后端,并设计结构化 CUDA-to-MLX 翻译层,让进化式内核搜索把已有 CUDA 内核当作知识库适配到 Apple Silicon。

推荐理由

K-Search 把 CUDA 内核优化经验迁移到 Apple Silicon,读者可了解跨平台内核搜索的方法与实测数据。

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