Member of Technical Staff – Post-Training, Reinforcement Learning, Generalization
Posted 59mins ago
Employment Information
Report this job
Job expired or something wrong with this job?
Job Description
Member of Technical Staff training Cohere’s enterprise foundation models with reinforcement learning. Developing generalization methods, evaluations, and production training recipes.
Responsibilities:
- Train large language models with reinforcement learning
- Develop methods that improve generalization beyond the training distribution
- Design training recipes, environments, and evaluations to assess learned skills
- Design, run, and analyze large-scale RL post-training experiments, including RLHF and RL from verifiable rewards
- Research reward design, environment and task diversity, curricula, and regularization
- Build held-out evaluations for unseen tasks, domains, and formats
- Detect 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 production-model training recipes
- Collaborate with post-training data infrastructure, evaluation, and pre-training teams
Requirements:
- Strong research and software engineering skills, with hands-on experience training LLMs
- Experience applying reinforcement learning to LLMs, such as PPO, GRPO, RLHF, or RLVR, ideally at scale
- Track record of work on generalization, robustness, or distribution shift through publications or shipped models
- Rigorous experimental practice, including clean ablations, careful controls, and sound conclusions from noisy results
- Experience with distributed training, such as JAX, PyTorch, FSDP, or Megatron, is a plus
- Excellent communication skills
- Deep understanding of data quality principles and best practices
- Willingness to dive deep into data and assess dataset quality and impacts on model training outcomes
Benefits:
- Offers equity
- Weekly lunch stipend of $75/£75 or equivalent in local currency
- Full health and dental benefits
- Separate budget for mental health
- RRSP matching, 401K, Pension Scheme
- 100% parental leave top-up for up to 6 months, for either parent
- Annual arts and culture, fitness/wellness, quality time, and workspace improvement benefits
- Education and learning stipend for conferences, courses, and coaching
- 6 weeks of paid vacation (30 working days)
- Budget for traveling to other offices if remote
- Annual company offsite
- Daily lunch program, snacks, and regular community and social events for office-based employees
- Co-working benefit for employees not near an office
- $500 home office stipend

















