Tune Gemma 3 1B in JAX with GRPO for reasoning (Part 6): Training Pipeline

Training Pipeline: Assembling the GRPO Engine