Berkeley AI Research·· 2026-07-29精选AI 评分73
K-Search 将 CUDA 优化经验迁移至 MLX，在 Apple Silicon 上实现最高约 20× Prefill 加速
From CUDA to MLX: How K-Search Brings Decades of Kernel Expertise to Apple Silicon
AI 导读
研究团队扩展了 K-Search，使其能借助结构化 CUDA-to-MLX 转换层，将 CUDA kernel 优化经验用于 Apple Silicon 上的 MLX kernel 搜索。
来源：Berkeley AI Research · bair.berkeley.edu