How the community trained Gemma to "Think" with Tunix and TPUs

Google hosted a Kaggle hackathon challenging developers to train non-reasoning Gemma-2-2B and Gemma-3-1B models into general reasoning models using Tunix and Kaggle TPUs. Over 11,000 entrants and 300+ submissions demonstrated that effective reasoning training is achievable with limited compute, with winning teams combining supervised fine-tuning, preference optimization, and reinforcement learning. The winning techniques, including G-RaR's rubric-based reward system and a three-stage pipeline using SimPO and GRPO, enable models to produce structured reasoning traces for complex tasks across key industries.

Large Language Models LLMs often benefit from "thinking" before they speak for complex tasks. Frontier LLMs like Gemini 3 and leading open weight models like Gemma 4 can produce explicit reasoning traces, commonly called Chain-of-Thought, before answering user questions. But how this reasoning capability is trained is often not disclosed. While there are many reasoning tutorials https://github.com/google/tunix/blob/main/examples/grpo gemma.ipynb available on the Internet to train for simple verifiable tasks such as math or coding, accessible and easy-to-reproduce training recipes including data, training strategy, runnable code and evaluations for general reasoning remain scarce. This motivated us to hold the Google Tunix Hack: Train a model to show its work https://www.kaggle.com/competitions/google-tunix-hackathon hackathon on Kaggle: we challenged developers to transform non-reasoning base models Gemma-2-2B and Gemma-3-1B into general reasoning models, using Tunix and Kaggle TPUs. The response was overwhelming: over 11,000 entrants and 300+ high-quality submissions proved that decent reasoning training can be done by the community even with a very limited compute budget Kaggle TPU v5e-8 for 9 hours . In this post, we’ll highlight the techniques used by the winners and share key recipes that allow models to reason across key vertical industries, so you can train your own reasoning models. Highlighting the Winners: Key Innovations The winning submissions demonstrated a sophisticated understanding of post-training, combining supervised learning, preference optimization, and reinforcement learning in creative ways. G-RaR trains Gemma models to produce structured reasoning by combining Supervised Fine-Tuning SFT with GRPO, driven by a novel rubric-based LLM-as-judge reward system. How It Improves Reasoning The model's reasoning power is improved by explicitly training it to "show its work" inside <reasoning tags before outputting an answer. The underlying technique for GRPO , G-RaR Rubrics as Rewards , uses a larger judge model Gemma-3-12B to evaluate the quality of these intermediate logical steps based on task-specific rubrics. By converting discrete rubric scores into continuous, normalized reward signals, the technique provides dense, smooth feedback on the model's logic. This allows the model to continuously improve its reasoning capabilities without relying solely on exact-match correctness, making it highly effective even for open-ended, non-verifiable tasks. Technical Solution The team utilized a two-stage post-training pipeline: Stage 1 SFT : The Gemma-2-2B-IT model is fine-tuned via LoRA on a ~33k sample dataset to establish a baseline. This "warm start" teaches the model to reliably output the <reasoning ...</reasoning <answer ...</answer structure. Stage 2 GRPO : The model is then refined using GRPO-based on a composite reward function Format Reward + Exact Answer Reward + G-RaR Score . To overcome compute constraints, the team used a split-mesh architecture on a single Kaggle TPU v5e-8, placing the policy/reference models on one mesh and the judge model on the other for true parallel execution. Evolving a 1B parameter model into a structured reasoning engine "Pinocchio" via a highly efficient, 9-hour TPU pipeline SFT → SimPO → GRPO How it Improves Reasoning The model learns to generate a structured <reasoning trace before answering, shifting from basic pattern matching to logical deduction. This is built sequentially: SFT instills foundational Chain-of-Thought, SimPO locks in strict formatting preventing verbosity hacks , and GRPO refines logic by using an LLM-as-a-Judge to reward coherence and heavily penalize hallucinations.. Technical Solution The pipeline consists of three stages: SFT Distillation : Trained on 70k prompts using an OSS-120B teacher model and a Gemini task-router. SimPO Alignment : Replaced memory-heavy DPO to efficiently enforce strict XML formatting. GRPO Refinement : Used Gemini 2.0 Flash as an asynchronous judge to dynamically reward accuracy, logic, and format. Customizing Tunix: The team explicitly extended the Tunix library to support this workflow by:- Injecting a custom SimPO loss function with length normalization into the DPOTrainer . - Creating a high-throughput, asynchronous evaluation engine to process GRPO reward signals on the fly. Distilling the structured " IDEA-E https://www.eapon.ca/wp-content/uploads/2017/03/12460 IDEA Framework Workbook final April 2013 .pdf " ethical reasoning framework into a 2B model using curriculum-guided GRPO and a fast TF-IDF reward system. Why it Improves Reasoning The IDEA-E scaffold forces the model through a step-by-step logical deduction before answering, preventing premature guessing. Simultaneously, the TF-IDF reward prevents verbose "yapping" by incentivizing the use of context-relevant vocabulary in the reasoning trace. Technical Solution The pipeline features two stages: SFT: Fine-tuning on teacher data to establish the IDEA-E format. GRPO: Reinforcement learning using curriculum guidance and a TF-IDF reward instead of a slow LLM judge. Customizing Tunix: The team extended Tunix by integrating their custom TF-IDF reward function into the Tunix GRPO pipeline, allowing for rapid, non-blocking reward calculations on the CPU. Honorable Mentions While the top three spots took the podium, several other submissions showcased strong creativity and technical depth: The Approach: Instead of relying solely on static offline datasets, they implemented an on-policy distillation https://thinkingmachines.ai/blog/on-policy-distillation/ method from scratch within the Tunix framework. They used a larger, highly capable teacher model trained in 3 phases to generate reasoning traces dynamically in response to the student model's generations during training, creating a tighter feedback loop. The Approach: Developed by participant TheItCrow , this project focused on custom dataset curation and structured reward modeling.- They curated the Deep-CoRGI Cognitive Reasoning Guided Interface dataset, specifically designed to teach Chain of Thought. - They trained a custom ThoughtTeacher reward model to evaluate not just the correctness of the final answer, but the logical flow of the reasoning steps themselves. We are also very impressed with several submissions that focus on reasoning training in specific domains, such as medical https://www.kaggle.com/competitions/google-tunix-hackathon/writeups/new-writeup-1767963319979 , chemistry https://www.kaggle.com/competitions/google-tunix-hackathon/writeups/introducing-gemmax , legal https://www.kaggle.com/code/rishirajacharya/grpo-gemma3-1b-for-legal-data-with-tunix and robotics https://www.kaggle.com/competitions/google-tunix-hackathon/writeups/tunix-robotics-reasoning-single-session . Medical : GRPO generates structured, step-by-step reasoning traces enhancing the interpretability and reliability of its complex clinical problem-solving outputs Chemistry : step-by-step reasoning traces benefited the chemistry use case by enabling a small language model to solve complex chemistry reasoning tasks. Legal : Post-training via GRPO reinforces structured, step-by-step reasoning, enabling the Gemma 3 1B model to accurately analyze complex legal data and produce reliable, logically sound interpretations. Robotics : step-by-step reasoning generation allows the model to solve multi-step robotics planning and decision-making tasks under single-session training constraints. Ready to build? The Tunix Hackathon democratizes training highly capable, structured reasoning models by producing so many impressive reasoning training recipes that are now all publicly available. With Tunix and free Kaggle TPUs, developers can now achieve strong results on accessible hardware. If you're ready to start post-training your own reasoning models, here are some resources to get started: Explore Tunix on GitHub: Check out the official Tunix repository https://github.com/google/tunix to access the code, documentation, and community examples. Try a Colab Tutorial: Spin up a free TPU instance in Google Colab and try the Tunix examples https://github.com/google/tunix/tree/main/examples to run your first SFT or RL loop. Learn More About Reinforcement Learning: Read the RL documentation https://tunix.readthedocs.io/en/latest/design.html rl in Tunix to understand how to leverage reinforcement learning to finetune your model.