Tunix
A JAX-native LLM Post-Training Library
...Tunix also leans into research ergonomics: logging, checkpointing, and metrics are built in, and the code is written to be hackable rather than monolithic. Overall it aims to shorten the path from an off-the-shelf base model to a well-aligned, task-ready model using scalable JAX primitives.