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

从 CUDA 到 MLX:K-Search 如何将数十年的内核专业知识带到 Apple Silicon

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

AI 导读

该研究介绍了 K-Search,一个利用 AI 优化 GPU 内核的框架,并将其扩展到 Apple Silicon 的 MLX 框架。通过将 CUDA 内核优化知识迁移到 Apple Silicon,实现了在 Apple Silicon 上达到接近专家级别的性能。

推荐理由

该研究展示了如何将 CUDA 内核优化知识迁移到 Apple Silicon,为跨平台 AI 编码工具提供了新的可能性。

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