Berkeley AI Research·精选AI 评分60
从 CUDA 到 MLX:K-Search 如何将多年内核优化经验带到 Apple Silicon
From CUDA to MLX: How K-Search Brings Decades of Kernel Expertise to Apple Silicon
AI 导读
IBM Research 基于 UC Berkeley Sky Lab 的 K-Search 演化式内核搜索框架,新增 MLX 后端和结构化 CUDA-to-MLX 翻译层,让已有 CUDA 内核作为知识库迁移到 Apple Silicon。
推荐理由
作者展示了把 CUDA 优化经验迁移到 Apple Silicon 的具体做法,并给出与原生 MLX 内核的对比数据。
来源:Berkeley AI Research · bair.berkeley.edu