Tunix 是 Google 推出的原生 JAX 后训练库，旨在消除训练多轮、使用工具的 LLM 推理代理时的 TPU 空闲瓶颈。它通过将高度并发、异步的 rollout 与解耦的 producer-consumer 流水线相结合，最大化硬件吞吐量，确保即使在代理等待网络 I/O 或环境步骤时，训练器也能持续获得数据。此外，Tunix 提供即插即用的抽象和持续的宏观级性能分析，使开发者能够轻松集成自定义开源环境，并对复杂的分布式工作流进行优化，而无需进行大规模代码重构。