跳到正文
Berkeley AI Research·精选AI 评分61

从 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 博客介绍给 K-Search 演进式内核搜索框架新增 MLX 后端和结构化 CUDA-to-MLX 翻译层,把已有 CUDA 内核知识迁移到 Apple Silicon。

推荐理由

研究展示 CUDA 优化经验可经结构化翻译迁移到 MLX,让演进式内核搜索在 Apple Silicon 上接近专家水平。

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