Senior/Staff Machine Learning Research Engineer, General Agents, Enterprise GenAI
About the role
Senior/Staff ML Engineer role on Scale AI's General Agents team, focusing on designing, building, and deploying production-ready AI agents for enterprise use cases, spanning model design, evaluation, deployment, and iteration with emphasis on generalization, extensibility, and customer deployment.
Responsibilities
- Design and implement end-to-end agent systems that combine LLM reasoning, tool use, memory, and control logic to solve recurring enterprise use cases.
- Build scalable, reliable agent architectures that can be deployed across many customers with varying data, tools, and constraints.
- Develop evaluation frameworks, datasets, environments, and metrics to measure agent performance, reliability, and business impact in production settings.
- Collaborate closely with product managers, customers, data annotators, and other engineering teams to translate enterprise requirements into robust agent designs.
- Productionize frontier agent techniques (e.g., planning, multi-step reasoning and tool-use, multi-agent patterns) into maintainable, observable systems.
- Own deployment, monitoring, and iteration of agent systems, including failure analysis and continuous improvement based on real-world usage.
- Contribute to technical direction and architectural decisions for general agent development best practices and methods, with increasing scope and leadership at the Staff level.
Qualifications
- 5+ years of experience building and deploying machine learning or AI systems for real-world, production use cases.
- Strong engineering fundamentals, supported by a Bachelor’s and/or Master’s degree in Computer Science, Machine Learning, AI, or equivalent practical experience.
- Deep understanding of modern LLMs, prompt-, context-, and system-level optimization, and agentic system design.
- Proven proficiency in Python, including writing production-quality, testable, and maintainable code.
- Experience building systems that integrate models with external tools, APIs, databases, and services.
- Ability to operate in ambiguous problem spaces, balancing research-driven approaches with pragmatic product constraints.
- Strong communication skills and comfort working in customer-facing or cross-functional environments.
Skills
Benefits
- comprehensive health, dental and vision coverage
- retirement benefits
- learning and development stipend
- generous PTO
- commuter stipend (optional/eligible)
About Scale AI
Data-centric infrastructure to accelerate the development of AI
Scale accelerates the development of AI within organizations of any size to deliver critical business insights and operational efficiency. Its data-centric infrastructure platform leverages RLHF (Reinforced Learning with Human Feedback) to help organizations build the strongest AI models that supercharge their business, with customers across industries including Meta, Microsoft, U.S. Army, DoD’s Defense Innovation Unit, Open AI, General Motors, Toyota Research Institute, Brex, Instacart and Flexport.
More open roles at Scale AI
- Strategic Contracts Lead, Public Sector4d agoWashington, DC · Public Sector
- Web Designer, Brand Web Experience4d agoSan Francisco, CA · Marketing
- Head of Strategic Partnerships, Global Systems Integrators5d agoSan Francisco, CA · Enterprise Sales
- Safety Research Scholar - Human Frontier Labs5d agoHybrid (United States) · Human Frontier Collective
- Machine Learning Research Intern (Summer 2027)5d agoHybrid (San Francisco, CA), Hybrid (New York, NY) · University