Berkeley AI Research·· 2026-07-29AI 评分49
从 CUDA 到 MLX：K-Search 如何将数十年内核专业经验引入 Apple Silicon
From CUDA to MLX: How K-Search Brings Decades of Kernel Expertise to Apple Silicon
AI 导读
研究团队为内核搜索框架 K-Search 拓展了 MLX 后端，通过结构化翻译层将 CUDA 优化经验自动迁移至 Apple Silicon。生成的 Attention 内核达到原生 MLX 0.97 倍性能，Mamba SSM 内核较社区 mlx-lm 实现取得最高 20 倍 prefill 加速。
来源：Berkeley AI Research · bair.berkeley.edu