Sakana AI 推出自动化 CUDA 优化智能体:PyTorch 代码自动转写加速
Sakana AI 推出一套自动化 CUDA 内核发现与优化框架及基准 robust-kbench。该框架能将 PyTorch 代码转写为 CUDA 内核,并通过演化生成机制和大模型验证器迭代优化底层性能。这意味着写底层算子优化的门槛大幅降低,繁重的工程调优可以交给 AI 自动完成。
Sakana AI 推出一套自动化 CUDA 内核发现与优化框架及基准 robust-kbench。该框架能将 PyTorch 代码转写为 CUDA 内核,并通过演化生成机制和大模型验证器迭代优化底层性能。这意味着写底层算子优化的门槛大幅降低,繁重的工程调优可以交给 AI 自动完成。