Google Tunix

Google Tunix (current version 0.1.6) is a lightweight, JAX-native framework designed for post-training Large Language Models (LLMs) using both reinforcement learning (RL) and supervised fine-tuning (SFT). It provides powerful tools for researchers and production teams to achieve maximum control and scalability when aligning and improving foundation models, particularly on accelerators like TPUs. Releases are frequent, focusing on new model support, API stability, and performance enhancements.