Senior/Staff Machine Learning Research Engineer, General Agents, Enterprise GenAI
$290k - $363k • Remote • San Francisco, CA; New York, NY
Posted 7d ago
Job Location
San Francisco, CA; New York, NY
Tech Stack
Remote Work Policy
Fully remote
Categories
AI Agent Engineer
About the job
Scale AI is seeking a Senior/Staff Machine Learning Engineer to join the General Agents team, focusing on building robust, production-ready AI agents for enterprise use cases. This role involves working across the entire agent lifecycle, from design and development to evaluation, deployment, and iteration. You will translate cutting-edge agentic techniques into reliable systems that solve high-impact business problems, emphasizing generalization, extensibility, and deployment across multiple customers. The team operates at the intersection of frontier agent development and real-world application, aiming to deliver significant economic value through scalable agent systems.
Responsibilities
- Design and implement end-to-end agent systems combining LLM reasoning, tool use, and memory.
- Build scalable and reliable agent architectures for diverse customer environments.
- Develop evaluation frameworks, datasets, and metrics to assess agent performance and impact.
- Collaborate with product managers, customers, and other engineering teams to define agent designs.
- Productionize advanced agent techniques like planning and multi-agent patterns into observable systems.
- Own the deployment, monitoring, and continuous improvement of agent systems.
- Contribute to technical direction and best practices for general agent development.
Requirements
- 5+ years of experience building and deploying production ML or AI systems.
- Bachelor's or Master's degree in Computer Science, Machine Learning, AI, or equivalent practical experience.
- Deep understanding of LLMs, prompt optimization, and agentic system design.
- Proficiency in Python, writing production-quality, testable code.
- Experience integrating models with external tools, APIs, and databases.
- Ability to navigate ambiguous problems, balancing research with product constraints.
- Strong communication skills and experience in customer-facing or cross-functional roles.
Benefits
- Base salary
- Equity compensation
- Comprehensive health, dental, and vision coverage
- Retirement benefits
- Learning and development stipend
- Generous PTO
- Commuter stipend (potential)