跳到正文
  1. Berkeley AI Research62

    从 CUDA 到 MLX:K-Search 如何把多年内核优化经验带到 Apple Silicon

    Berkeley Sky Lab 与 IBM Research 的研究者把 K-Search 内核搜索框架扩展到 MLX,并加入结构化的 CUDA 到 MLX 翻译层,让已有 CUDA 内核作为知识库被改写为 Apple Silicon 内核。

    推荐理由:作者展示了把 CUDA 优化经验迁移到 MLX 的具体做法,并给出注意力与 Mamba 两个内核的实测对比。

已经到底了