polish: detailed training notebook (drop optional, add reward demo + comparison)

Update notebooks/train_grpo.ipynb

add training notebook for hackathon submission