GitHub · 灵魂体仓库·· 20 天前AI 评分51
Google 发布 JAX 后训练库 Tunix,支持 SFT、RL 与 Agentic RL
google/tunix
来源与原文
GitHub · 灵魂体仓库
发布时间:(北京时间)
阅读原始来源 · github.com ↗完整链接与内容说明
https://github.com/google/tunix
本站目前仅有 AI 导读与摘要,未展示原文正文;技术细节、代码和完整结论请查看原始来源。
AI 导读
Google 发布了基于 JAX 的后训练库 Tunix(Tune-in-JAX),目前处于 V2 发布状态,支持 SFT、PPO、GRPO 等强化学习以及多轮工具使用和异步 rollout 的 Agentic RL。该库与 Flax NNX、MaxText 集成,并通过 vLLM 和 SGLang-JAX 在 TPU 上做推理 rollout,支持 Gemma、Llama、Qwen 模型族,可借助 Pathways 扩展到数千设备的多主机分布式训练。
来源:GitHub · 灵魂体仓库 · github.com