Member of Technical Staff - Post-Training (Reinforcement Learning & Generalization)

📋 Description

• Design, run, and analyse large-scale RL post-training experiments (e.g., RLHF, RL from verifiable
• Research and develop methods that improve out-of-distribution generalization, such as reward
• Build held-out evaluations that measure generalization to unseen tasks, domains, and formats, and
• Diagnose training dynamics and failure modes, including instability, reward over-optimization, and
• Turn successful research into robust, reproducible training recipes for production models.

🎯 Requirements

• Strong research and software engineering skills, with hands-on experience training LLMs.
• Experience applying reinforcement learning to LLMs (e.g., PPO, GRPO, RLHF, RLVR), ideally at scale.
• Track record of work on generalization, robustness, or distribution shift, through publications or
• Rigorous experimental practice: clean ablations, careful controls, and sound conclusions from noisy
• Experience with distributed training (e.g., JAX, PyTorch, FSDP, Megatron) is a plus.
• Excellent communication skills.

🎁 Benefits

• A weekly lunch stipend of $75/£75 or equivalent in your local currency for lunch.
• Full health and dental benefits, including a separate budget for mental health.
• RRSP matching, 401K, Pension Scheme.
• 100% Parental Leave top-up for up to 6 months, for either parent.
• Annual enrichment benefits: Arts & culture, fitness/wellness, quality time, and a workspace
• Education & learning stipend for conferences, courses, and coaching.