Tunixは、Googleの新しいJAXネイティブのポストトレーニングライブラリであり、マルチターンのツール使用LLM推論エージェントのトレーニングにおけるTPUアイドル時のボトルネックを解消するように設計されています。これは、高度に並行した非同期ロールアウトと、分離されたプロデューサー・コンシューマーパイプラインを組み合わせることでハードウェアのスループットを最大化し、エージェントがネットワークI/Oや環境ステップを待機している間でもトレーナーに常にデータが供給されるようにします。さらに、Tunixはプラグアンドプレイの抽象化と継続的なマクロレベルのプロファイリングを提供し、開発者がカスタムオープンソース環境を簡単に統合し、大規模なコードの書き直しなしに複雑な分散ワークフローを最適化できるようにします。