Berkeley AI Research·精选AI 评分78
Berkeley AI Research：K-Search 将 CUDA 内核优化经验迁移至 Apple Silicon
From CUDA to MLX: How K-Search Brings Decades of Kernel Expertise to Apple Silicon
AI 导读
Berkeley AI Research 扩展了 K-Search 框架，新增 MLX 后端以支持 Apple Silicon，并通过结构化 CUDA-to-MLX 翻译层将 NVIDIA 内核优化知识迁移至苹果芯片。
推荐理由
文章展示了如何将 CUDA 内核优化知识通过结构化翻译层迁移至 Apple Silicon，使 MLX 框架下的 Attention 和 Mamba SSM 内核达到接近专家级的性能。
来源：Berkeley AI Research · bair.berkeley.edu