Google 发布基于 JAX 的智能体后训练库 Tunix

OmniTools 7月22日消息，Google 7月21日消息，Google 正式发布 Tunix——一个基于 JAX 的原生后训练库，专为多轮、工具增强型大语言模型（LLM）推理智能体设计。该库聚焦解决训练过程中 TPU 硬件闲置瓶颈问题。

Tunix 采用高并发异步 rollout 机制与解耦的生产者-消费者流水线架构，显著提升硬件吞吐量，保障训练器持续获取数据流。

库内提供即插即用的抽象接口及持续宏观级性能分析能力，支持开发者便捷集成自定义环境。