Google 发布 Tunix,一个基于 JAX 的原生后训练库,旨在消除多轮、使用工具的 LLM 推理智能体训练中的 TPU 闲置瓶颈。Tunix 通过高并发异步 rollout 与解耦的生产者-消费者流水线最大化硬件吞吐量,确保训练器持续获得数据。该库提供即插即用抽象和持续宏观级性能分析,便于集成自定义环境。
产品
·Google Developers Blog(RSS)
Google 推出 Tunix:基于 JAX 的高吞吐智能体后训练库
— Scaling Agentic RL: High-Throughput Agentic Training with Tunix