Sr. Staff AI/ML Engineer

The Wrap; California; Hybrid; USD 288,000 - 321,000; Full time

Job summary

Gusto in the United States is seeking a Staff Machine Learning Engineer to design, build, and scale the AI platform powering internal teams. You will own lifecycle from problem framing to production deployment, focusing on agent orchestration, RAG pipelines, and model serving to enable rapid, reliable AI-powered experiences.

You will mentor engineers, set technical standards, collaborate across product, design, and PMs, and stay at the forefront of AI/ML research.

Qualifications

• 12+ years building end-to-end AI/ML systems.
• Experience designing platform-level infrastructure used by other engineering teams.
• Hands-on with LLM-based apps and agentic workflows.
• Strong Python and software engineering fundamentals.
• Ability to lead cross-functional teams and communicate with non-technical stakeholders.
• Ph.D. or Master’s in CS/ML is a plus.

Responsibilities

• Design and build scalable platform services for AI products.
• Lead technical strategy and architecture for the AI platform.
• Collaborate with app teams, PMs, and Designers to understand AI needs.
• Build, harden, and operate shared infrastructure powering intelligent agents.
• Stay current with AI/ML research and prototype new techniques.
• Establish validation, deployment, monitoring, and performance practices.
• Communicate strategy to executives and non-technical partners.
• Mentor engineers across teams.

Skills

LLMs & NLP

Retrieval & RAG

Agent orchestration

Deep learning

Reinforcement learning

Python

CI/CD & testing

Cloud platforms

Communication

Technical leadership

Education

Ph.D. or Master’s in CS/ML/Statistics/Math

Tools

PyTorch

TensorFlow

Hugging Face

GCP

AWS

Azure

Job description

About Gusto

At Gusto, we're on a mission to grow the small business economy. We handle the hard stuff — payroll, health insurance, 401(k)s, and HR — so owners can focus on their craft and their customers. With teams in Denver, San Francisco, and New York, we support more than 500,000 small businesses nationwide and are building a workplace that reflects the people we serve.

All full-time employees receive competitive base pay, benefits, and equity (RSUs) — because everyone who helps build Gusto should share in its success. Offer amounts are determined by role, level, and location. Learn more about our Total Rewards philosophy.

AI is a fundamental part of how work gets done at Gusto. We expect all team members to actively engage with AI tools relevant to their role and grow their fluency as the technology evolves. AI experience requirements vary by role and will be assessed during the interview process.

About the Role:

• You'll design, build, and scale the AI platform that empowers Gusto's internal teams to build intelligent agents and deliver customer value. This includes developing core platform capabilities — agent orchestration, RAG infrastructure, eval tooling, model serving, prompt management, observability, and safety guardrails — that enable product teams to rapidly build, test, and ship AI-powered experiences. You'll own the full lifecycle from problem framing to production deployment, ensuring the platform is reliable, performant, and easy to adopt. As a Staff MLE, you'll also shape technical standards, drive architectural decisions, and mentor engineers across the organization.

About the Team:

• You'll join the CoreX AI Platform team, responsible for building the foundational AI infrastructure that Gusto's internal app teams rely on to create and ship intelligent agents. We're a small, high-impact group of ML engineers and platform builders working closely with Product Engineers, PMs, and Designers across the company. We move fast, prototype boldly, and focus on making it easy for any team a

What you'll do day-to-day:

• Design and build scalable platform services — agent orchestration, RAG pipelines, eval frameworks, model serving — that internal teams use to ship AI-powered products.
• Lead technical strategy and architecture for the AI platform, including model lifecycle, observability, safety guardrails, and evaluation infrastructure.
• Collaborate cross-functionally with app teams, PMs, and Designers to understand their AI needs and deliver platform capabilities that unblock them.
• Build, harden, and operate shared infrastructure that powers intelligent agents across Gusto (e.g., retrieval, routing, prompt management, content selection).
• Stay current with AI/ML research; rapidly prototype and productionize new techniques that strengthen the platform.
• Establish robust practices for validation, deployment, monitoring, and ongoing performance management of platform services and the agents built on them.
• Communicate technical strategy, tradeoffs, and impact clearly to executives and non-technical partners.
• Mentor engineers across teams, raising the bar on AI/ML best practices and fostering a culture of pragmatic innovation.

Here’s what we’re looking for

• 12+ years building and deploying end-to-end AI/ML systems, with experience designing platform-level infrastructure that other engineering teams build on.
• Deep expertise in one or more areas: LLMs, NLP, retrieval/RAG, agent orchestration, deep learning, or reinforcement learning.
• Hands-on experience building LLM-based applications and agentic workflows — including prompt engineering, retrieval design, evaluation, and production deployment.
• Proficiency with modern ML frameworks (PyTorch, TensorFlow, Hugging Face) and cloud platforms (GCP, AWS, or Azure).
• Strong Python skills and sound software engineering fundamentals — testing, code review, CI/CD, reliability, and API design.
• Proven track record shipping impactful AI/ML projects to production, ideally in a platform or infrastructure context.
• Demonstrated ability to lead cross-functionally, influence technical direction, and communicate clearly with both engineers and non-technical stakeholders.
• Ph.D. or Master's in CS, ML, Statistics, Mathematics, or related field is a plus.

Compensation

Our cash compensation amount for this role is targeted at $245,000-272,000 in Denver & most remote locations, and $288,000-321,000 for San Francisco & New York. Final offer amounts are determined by multiple factors including candidate experience and expertise and may vary from the amounts listed above.

Gusto has physical office spaces in Denver, San Francisco, and New York City. Employees who are based in those locations will be expected to work from the office on designated days approximately 2-3 days per week (or more depending on role). The same office expectations apply to all Symmetry roles, Gusto's subsidiary, whose physical office is in Scottsdale.

Note: The San Francisco office expectations encompass both the San Francisco and San Jose metro areas.

When approved to work from a location other than a Gusto office, a secure, reliable, and consistent internet connection is required. This includes non-office days for hybrid employees.

Our customers come from all walks of life and so do we. We hire great people from a wide variety of backgrounds, not just because it’s the right thing to do, but because it makes our company stronger. If you share our values and our enthusiasm for small businesses, you will find a home at Gusto.

Gusto is proud to be an equal opportunity employer. We do not discriminate in hiring or any employment decision based on race, color, religion, national origin, age, sex (including pregnancy, childbirth, or related medical conditions), marital status, ancestry, physical or mental disability, genetic information, veteran status, gender identity or expression, sexual orientation, or other applicable legally protected characteristic. Gusto considers qualified applicants with criminal histories, consistent with applicable federal, state and local law. Gusto is also committed to providing reasonable accommodations for qualified individuals with disabilities and disabled veterans in our job application procedures. We want to see our candidates perform to the best of their ability. If you require a medical or religious accommodation at any time throughout your candidate journey, please fill out this form and a member of our team will get in touch with you.

Gusto takes security and protection of your personal information very seriously. Please review our Fraudulent Activity Disclaimer.

Personal information collected and processed as part of your Gusto application will be subject to Gusto's Applicant Privacy Notice.