Google Tunix Hack - Train a model to show its work

Teach a LLM to reason using Tunix, Google’s new JAX-native library for LLM post-training.