Staff Machine Learning Research Engineer (Enterprise GenAI)

TL;DR

Staff Machine Learning Research Engineer (Enterprise GenAI): Build and optimize next-generation Agent RL training platforms for enterprise AI applications with an accent on post-training algorithms and multi-agent systems. Focus on designing and integrating cutting-edge reinforcement learning methods and training state-of-the-art models for complex enterprise use-cases.

Location: San Francisco, CA or New York, NY, USA (onsite)

Salary: $180,600–$315,000 USD

Company

Scale AI accelerates AI development by providing high-quality data and full-stack technologies powering leading AI models for enterprises and governments worldwide.

What you will do

• Train state-of-the-art models developed internally and from the community for enterprise deployment.
• Research and integrate cutting-edge algorithms into the training stack.
• Design solutions enabling complex multi-agent systems to learn from process and outcome-based rewards.
• Collaborate with MLREs on the Enterprise AI team to deploy diverse AI use-cases including cybersecurity and healthtech models.

Requirements

• Location: Must be based in San Francisco, CA or New York, NY (onsite)
• 5+ years of LLM training experience in production environments.
• Experience with post-training methods such as RLHF, RLVR, PPO, GRPO.
• Recent publications in top ML conferences (NEURIPS, ICLR, ICML).
• PhD or Masters in Computer Science or related field.
• English: B2 or higher proficiency required

Culture & Benefits

• Comprehensive health, dental, and vision coverage.
• Retirement benefits and equity compensation.
• Learning and development stipend.
• Generous paid time off and commuter stipend.
• Inclusive and equal opportunity workplace with accommodations for disabilities.