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

从 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 团队为 K-Search 演化式内核搜索框架新增 MLX 后端,并设计结构化 CUDA-to-MLX 翻译层,让搜索以现有 CUDA 内核为知识库生成 Apple Silicon 内核。

推荐理由

作者把 CUDA 优化知识整理成结构化上下文,让演化搜索在 Apple Silicon 上接近专家内核水平,方法可迁移到其他硬件生态。

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