Tunix — это новая библиотека Google на базе JAX, предназначенная для устранения узких мест простоя TPU при обучении многоходовых LLM-агентов, использующих инструменты. Она максимизирует пропускную способность оборудования, сочетая высококонкурентные асинхронные развертывания с отдельным конвейером продюсер-потребитель, гарантируя, что трейнер постоянно получает данные, даже когда агенты ожидают сетевого ввода-вывода или шагов среды. Кроме того, Tunix предоставляет абстракции, которые можно подключать и использовать, а также непрерывное профилирование на макроуровне, позволяя разработчикам легко интегрировать пользовательские среды с открытым исходным кодом и оптимизировать сложные распределенные рабочие процессы без масштабных переписываний кода.