Berkeley AI Research·· 2026-07-29精选AI 评分60
从 CUDA 到 MLX:K-Search 如何把数十年内核经验带到 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 进化式内核搜索框架扩展到 MLX,通过结构化 CUDA-to-MLX 翻译层把已有 CUDA 内核作为知识库,在 Apple Silicon 上生成高质量内核。
推荐理由
K-Search 把 CUDA 内核优化经验迁移到 Apple Silicon,给出注意力与 Mamba 两类内核的实测数据,可看跨平台迁移方法。
来源:Berkeley AI Research · bair.berkeley.edu