Member of Technical Staff – Post-Training, Reinforcement Learning, Generalization

🔥 3 minutes ago

🏄 California, New York, +1 more states – Remote

🏄 California – Remote 🗽 New York – Remote ☕ Washington – Remote

💵 CA$295k - CA$535k / year

⏰ Full Time

🔴 Lead

🖥 Software Engineer

🦅 H1B Visa Sponsor

This company has sponsored H1B visas in the past.

👻 Ghost score 21%

What it means: how likely this posting is sitting open without active hiring.How to read it: 60% or higher means we think this is a ghost. Lower scores are less of a worry.How we score it: age, reposts, open roles vs team size, how fast jobs close, and whether pay is listed.Fine print: an estimate from public data, not a judgment of the company. Real jobs can score high, so treat it as a hint, not a fact.

Apply Now

Find Similar Remote Jobs

Set Up Email Alert For New Jobs

📊 Check your resume score for this job

Improve your chances of getting an interview by checking your resume score before you apply.

Upload Resume

Cohere

WebsiteLinkedInAll Job Openings

11 - 50 employees

🤖 Artificial Intelligence

🏢 Enterprise

☁️ SaaS

Artificial Intelligence • Enterprise • SaaS

Cohere is a leading enterprise AI platform optimized for generative AI, search and discovery, and advanced retrieval. The company offers AI-powered applications designed to augment and elevate the global workforce, helping businesses thrive in the AI era. Cohere provides solutions such as embedding and reranking models, allowing enterprises to efficiently retrieve information and build powerful applications. The company offers flexible deployment options for enterprise-grade AI, on any cloud or on-premises, and provides extensive developer resources and support. Cohere is committed to scaling intelligence to serve humanity, making intelligence abundant, affordable, and accessible.

📋 Description

• 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