开源编译器 Latch 从 PyTorch 生成训练 megakernel,H100 上提速 2.75 倍

sguada · x · 2026-10-27

有团队融资 10 亿美元开发面向数据高效学习的 megakernel,而独立研究者 deepaksinghml 没有这笔钱,于是开源了 Latch:一个直接从 PyTorch 代码生成训练 megakernel 的编译器。

作者公布的 H100 实测数据(vs Compiled PyTorch):

项目已开源,定位是用编译器自动化 megakernel 生成,让没有巨额资金的小团队也能做内核级训练优化。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →