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