Member of Technical Staff - Post-Training (Reinforcement Learning & Generalization)
RemoteCA$295k - CA$535kFull-timePosted 1w ago
About the role
Role focused on training large language models with reinforcement learning and improving generalization beyond training distribution. Design training recipes, environments, and evaluations; collaborate with post-training data infrastructure, evaluation, and pre-training teams.
Responsibilities
- Design, run, and analyse large-scale RL post-training experiments (e.g., RLHF, RL from verifiable rewards)
- Research and develop methods that improve out-of-distribution generalization, such as reward design, environment and task diversity, curricula, and regularization
- Build held-out evaluations that measure generalization to unseen tasks, domains, and formats, and that catch reward hacking and benchmark overfitting
- Diagnose training dynamics and failure modes, including instability, reward over-optimization, and capability regressions
- Turn successful research into robust, reproducible training recipes for production models
Qualifications
- 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 shipped models
- Rigorous experimental practice: clean ablations, careful controls, and sound conclusions from noisy results
- Experience with distributed training (e.g., JAX, PyTorch, FSDP, Megatron) is a plus
- Excellent communication skills
- Deep understanding of data quality principles and best practices: demonstrates willingness to dive deep into data while possessing strong intuition for assessing the quality of datasets and anticipating how different data characteristics will impact model training outcomes
Skills
Required:Reinforcement Learning for LLMsGeneralization & RobustnessExperimental Methodology
Nice to have:Distributed Training
Benefits
- Weekly lunch stipend
- Full health and dental benefits, including mental health budget
- RRSP matching, 401K, Pension Scheme
- 100% Parental Leave top-up for up to 6 months
- Annual enrichment benefits (arts & culture, fitness/wellness, quality time, workspace improvement credit)
- Education & learning stipend for conferences, courses, and coaching
- 6 weeks of paid vacation (30 working days)
- Budget for traveling to other offices if remote, plus annual company offsite
- $500 home office stipend
Last updated Oct 21, 2026. Interested? Apply on the company site.
About Cohere
Support automation that's actually intelligent
Cohere resolves 40-70% of support requests using AI, over all channels, within just a few days.
artificial-intelligenceautomationliquid-ai-recruiting-target
More open roles at Cohere
- Partner Development Manager - Berlin5d agoRemote · Revenue
- VP of Sales - Public Sector5d agoRemote · Revenue
- Delivery Operations Analyst5d agoRemote · Product Management & Program Management
- Technical Program Manager, AI Delivery for Public Sector & Defense, Germany5d agoRemote · Product Management & Program Management
- Senior/Staff Software Engineer (Back-End), Inference5d agoRemote · Inference