研究者开源 theseus:用人类语言做神经网络架构研究

pratyusha_PS · x · 2026-09-15

作者 houjunliu 开源了自己做架构研究用的系统 theseus(GitHub 仓库 Jemoka/thx)。这是一个模板仓库,克隆即用,基于 uv 管理,支持 CUDA 12/13、TPU 和 CPU 等多种后端。

特点是让架构研究可以直接用「人类语言」描述驱动 GPU/TPU 训练;性能不算激进,中位数天气下约 20 多 MFU,状态好时 30 多 MFU,胜在简单可用,作者欢迎他人使用并反馈。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →