跳到正文
Berkeley AI Research·· 2026-07-29AI 评分47

从 CUDA 到 MLX:K-Search 如何将内核优化经验引入 Apple Silicon

From CUDA to MLX: How K-Search Brings Decades of Kernel Expertise to Apple Silicon

AI 导读

扩展后的内核搜索框架 K-Search 引入结构化 CUDA 到 MLX 转译层,能将成熟的 GPU 内核优化经验自动迁移至 Apple Silicon。该系统由 Gemini 3.5 Pro Preview 执行优化推理并生成代码,在真实硬件上闭环迭代。

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