00:08
elvis@omarsar0
elvis@omarsar0精选
推荐理由：想又快又稳地优化CUDA内核？Kernel Forge用MCTS自动探索，比手动写靠谱多了，实测14个内核超基线。
Kernel Forge
共 1 条相关 AI 资讯
7月30日
00:08
Kernel Forge是一个开源agent harness，能直接改写PyTorch模型的CUDA内核。它覆盖视觉、扩散和LLM三类工作负载。采用蒙特卡洛树搜索(MCTS)而非线性生成-修复链，并提供GUI监控进度。在NVIDIA DGX Spark（GB10 GPU）上，经过每核50次优化迭代，它在四个模型中优化了14个内核，使其超越PyTorch基线。性能提升主要来自harness结构和原地重整合，而非更强的LLM。