Berkeley AI Research·· 2026-07-29精选AI 评分65
K-Search 扩展 MLX 后端:将 CUDA 内核优化知识迁移至 Apple Silicon
From CUDA to MLX: How K-Search Brings Decades of Kernel Expertise to Apple Silicon
AI 导读
Berkeley AI Research 与 IBM Research 将 K-Search 进化式内核搜索框架扩展至 Apple 的 MLX 框架,通过结构化 CUDA-to-MLX 翻译层,让 AI 自动生成适配 Apple Silicon 的高性能 GPU 内核。
推荐理由
原文给出了CUDA优化知识向MLX迁移的具体方法和实测数据,读者可据此评估跨硬件内核优化的可行性。
来源:Berkeley AI Research · bair.berkeley.edu