Berkeley AI Research·· 2026-07-29精选AI 评分60
K-Search 从 CUDA 迁移到 MLX,为 Apple Silicon 生成高性能内核
From CUDA to MLX: How K-Search Brings Decades of Kernel Expertise to Apple Silicon
AI 导读
Berkeley Sky Lab 与 IBM Research 将 K-Search 内核搜索框架扩展到 MLX,通过结构化的 CUDA-to-MLX 翻译层让已有 CUDA 内核经验适配 Apple Silicon。
推荐理由
读者可以看到 CUDA 内核经验如何被结构化翻译成 MLX 优化策略,以及翻译层对最终性能的具体贡献。
来源:Berkeley AI Research · bair.berkeley.edu