Berkeley AI Research·· 2026-07-29AI 评分58
Berkeley 团队用 K-Search 将 CUDA 内核优化经验迁移至 Apple Silicon,MLX Attention 达原生 0.97x、Mamba prefill 提速约 20x
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。
来源:Berkeley AI Research · bair.berkeley.edu