Tunix는 Google의 새로운 JAX 기반 후처리 라이브러리로, 멀티턴, 도구 사용 LLM 추론 에이전트 훈련 시 TPU 유휴 병목 현상을 제거하도록 설계되었습니다. 이는 고도로 동시적이고 비동기적인 롤아웃과 분리된 생산자-소비자 파이프라인을 결합하여 하드웨어 처리량을 극대화하며, 에이전트가 네트워크 I/O 또는 환경 단계를 기다리는 동안에도 트레이너가 지속적으로 데이터를 공급받도록 보장합니다. 또한 Tunix는 플러그 앤 플레이 추상화와 지속적인 매크로 수준 프로파일링을 제공하여 개발자가 사용자 정의 오픈 소스 환경을 쉽게 통합하고 대규모 코드 재작성 없이 복잡한 분산 워크플로우를 최적화할 수 있도록 합니다.