跳到正文
四海AI汇/文章阅读
Berkeley AI Research·· 2026-07-29精选AI 评分62

从 CUDA 到 MLX:K-Search 如何把多年 kernel 经验带到 Apple Silicon

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

AI 导读

研究团队把 K-Search 的进化式 kernel 搜索扩展到 MLX 后端,并加入结构化的 CUDA 到 MLX 翻译层,让已有 CUDA kernel 经验直接迁移到 Apple Silicon。

推荐理由

读者可以看到 CUDA 优化经验如何被结构化翻译成 MLX 策略,以及翻译层对搜索结果的量化影响。

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