Oossa

TorchTitan 新增模型支持与强化学习功能

PyTorch 的 TorchTitan 平台现已支持最新开放权重模型,并推出统一的强化学习技术栈 TitanRL。

简讯作者: Oossa 发布日期: 1 分钟阅读

PyTorch 原生训练平台 TorchTitan 宣布推出多项升级。平台现已支持 Muse Spark Open、Qwen3‑VL、Kimi K2.7、Kimi K3 和 DeepSeek‑V4 等最新开放权重模型,并将它们接入其五维并行体系。此次更新还加入了强化学习框架 TitanRL,可在预训练和推理中复用完全相同的预训练模型,无需维护独立代码库。性能方面的改进包括更快的内核、更完善的内存管理,以及可加速大规模训练的全新通信功能。

为什么重要

研究人员可以采用最新模型并开展强化学习实验,无需重写代码,从而节省时间并减少错误。

这篇文章有帮助吗?
分享

继续阅读

Oossa · 新闻简报

一周 AI,讲明白

每周一,用通俗的语言讲清值得知道的新闻。免费,无垃圾邮件。