Scale AI logo

Senior/Staff Machine Learning Research Engineer (General Agents, Enterprise GenAI)

Scale AI

On-siteSan Francisco, CAlead$265k–$331kPosted 8h ago

Job description

  • The General Agents team, part of Scale’s Enterprise organization, builds robust general agents for customer use cases and applications

  • The team sits at the intersection of frontier agent development and real-world deployment, translating state-of-the-art reasoning and agentic capabilities into reliable, production-grade systems that drive real economic value

  • Our agents are scalable systems built around recurring enterprise problem domains, with a strong emphasis on generalization, extensibility, and deployment across many customers

  • As a Senior/Staff Machine Learning Engineer (MLE) on the General Agents team, you’ll play a critical role in designing, building, and deploying production-ready AI agents that solve high-impact enterprise problems

  • You will work across the full agent lifecycle—from model and system design to evaluation, deployment, and iteration—bridging cutting-edge agentic techniques with the constraints and requirements of real customer environments

  • 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

Benefits

  • Health & Wellbeing: Our holistic approach to supporting Scaliens includes comprehensive health coverage, dental and vision insurance, mental healthcare services, and more. PTO policies and accommodating schedules ensure you’ll get time off when you need it to relax and recharge. Note that our offerings may vary by region as we strive to respond to the unique needs of Scaliens around the globe.

  • Personal & Career Growth: Continuously learn and grow through annual learning & development stipend, attending leadership breakfasts, manager training, speaker series, and joining an ERG.

  • Building Scale Community: We welcome guests to our offices, and you can expect to see Scalien families and friends around. Join local happy hours, and accept invites to game nights, book clubs, and many other employee-led community events.

  • Parental Support: Balancing work and family is essential, and Scale understands the importance of having adequate leave policies in place to promote a healthy home and work life.- 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

  • Experience building systems that integrate models with external tools, APIs, databases, and services

  • Deep understanding of modern LLMs, prompt-, context-, and system-level optimization, and agentic system design

  • Strong engineering fundamentals, supported by a Bachelor’s and/or Master’s degree in Computer Science, Machine Learning, AI, or equivalent practical experience

  • 5+ years of experience building and deploying machine learning or AI systems for real-world, production use cases

  • Proven proficiency in Python, including writing production-quality, testable, and maintainable code

  • Hands-on experience building AI agents using modern generative AI stacks (OpenAI APIs, commercial or open-source LLMs)

  • Experience with agent frameworks, orchestration layers, or workflow systems (e.g., tool calling, planners, multi-agent setups)

  • Familiarity with evaluation, monitoring, and observability for LLM-powered systems in production

  • Experience deploying ML systems in cloud environments and operating them at scale

  • Experience fine-tuning or adapting foundation models using methods like supervised fine-tuning (SFT), reinforcement learning with verifiable rewards (RLVR), and low-rank adaptation (LoRA) to improve agent performance on domain-specific tasks

  • Interest in shaping the future of general-purpose enterprise agents and their real-world impact