Berkeley AI Research·· 2026-07-29精选AI 评分62
K-Search 新增 MLX 后端,把 CUDA 内核经验迁移到 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 上的 Metal/MLX 内核。
推荐理由
K-Search 的 MLX 后端已开源,在 Apple Silicon 上做本地推理的可以按 README 复现 Attention 与 Mamba 内核的搜索流程。
来源:Berkeley AI Research · bair.berkeley.edu