Machine Learning Research Engineer, Agent Data Foundation - Enterprise GenAI
$265k - $331k • San Francisco, CA; New York, NY
Posted 2mo ago
About the job
Scale is seeking a Machine Learning Research Engineer to join the Agent Data Foundation team within the Enterprise ML Research Lab. This role will focus on defining and building the data flywheel that drives AI development, particularly for complex agents in enterprise settings. You will conduct cutting-edge research in areas such as synthetic data generation for RL post-training, building agents for production trace analysis, and contributing to a framework for agent construction. The goal is to create state-of-the-art agents through advanced post-training and agent-building algorithms, shaping the future of the GenAI movement.
Responsibilities
- Build synthetic data pipelines to generate enterprise environments for RL post-training.
- Create agents to convert production traces into actionable insights for agent improvement.
- Contribute to an agent building product that constructs other agents using coding agents and proprietary algorithms.
- Train state-of-the-art models, both internal and community-developed, for deployment to enterprise customers.
Requirements
- 3+ years of experience building with LLMs in a production environment.
- Clear experience constructing high-quality data to improve LLMs/Agents.
- Publications in top conferences such as NEURIPS, ICLR, or ICML within the last two years.
- PhD or Masters in Computer Science or a related field.
Benefits
- Base salary
- Equity
- Comprehensive health, dental and vision coverage
- Retirement benefits
- Learning and development stipend
- Generous PTO
- Commuter stipend