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

从 CUDA 到 MLX:K-Search 如何将内核优化经验迁移到 Apple Silicon

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

AI 导读

伯克利 Sky Lab 团队把 K-Search 内核搜索框架扩展到 Apple MLX,用结构化 CUDA-to-MLX 翻译层让已有的 CUDA 内核充当知识库,在 Apple Silicon 上生成高质量内核。

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