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

K-Search 扩展 MLX 后端,将 CUDA 内核经验迁移至 Apple Silicon

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 翻译层,把已有 CUDA 内核作为知识库改写为 Apple Silicon 上的高质量内核,而非逐条指令照搬。

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