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

Berkeley Sky Lab 团队用 K-Search 将 CUDA 内核优化经验迁移到 Apple Silicon 的 MLX 框架

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 翻译层,将 NVIDIA 生态多年积累的内核优化知识自动迁移到 Apple Silicon。

推荐理由

原文给出可复现的方法和具体数字,说明结构化 CUDA 翻译层在 Apple Silicon 上带来的性能差异。

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