Tunix 是谷歌推出的全新基于 JAX 的原生后训练库，旨在消除训练多轮、使用工具的大语言模型推理智能体时，因 TPU 闲置导致的瓶颈。它通过将高度并发的异步推演与解耦的生产者-消费者流水线相结合，最大化硬件吞吐量，确保即使在智能体等待网络 I/O 或环境步骤时，训练器也能持续获得数据供给。此外，Tunix 还提供即插即用的抽象层和持续的宏观级性能分析，使开发者能够轻松集成自定义的开源环境，并优化复杂的分布式工作流，而无需进行大量的代码重写。