Berkeley AI Research·· 2026-07-29精选AI 评分60
K-Search 将 CUDA 内核优化知识迁移至 Apple Silicon 的研究
From CUDA to MLX: How K-Search Brings Decades of Kernel Expertise to Apple Silicon
AI 导读
Berkeley AI Research 发布研究，介绍 K-Search 框架如何通过结构化 CUDA-to-MLX 翻译层，将 NVIDIA 生态的内核优化知识自动迁移至 Apple Silicon 平台。
推荐理由
研究展示了通过结构化翻译层将 CUDA 优化知识迁移至 Apple Silicon 的方法，在 Mamba SSM 内核上实现了约 20 倍的预填充加速。
来源：Berkeley AI Research · bair.berkeley.edu