跳到正文
原文
Berkeley AI Research·· 2026-07-29AI 评分41

从 CUDA 到 MLX:K-Search 如何将数十年 kernel 经验带到 Apple Silicon

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

AI 导读

Berkeley Sky Lab 将进化式 kernel 搜索框架 K-Search 扩展到 MLX 后端,通过结构化 CUDA-to-MLX 翻译层,把已有 CUDA kernel 当作知识库适配为 Apple Silicon 上的高质量 kernel。

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