跳到正文
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