Software Engineer, ML Platform
Build and scale Gusto’s machine learning and AI platform, including MLOps pipelines, model deployment frameworks, infrastructure, and observability. The role requires 5+ years of software engineering experience, proficiency in Python, Ruby, or Java, and experience with ML lifecycle infrastructure and cloud platforms.
About the job
Responsibilities
- Build core components of the ML and AI platform technical roadmap, including MLOps solutions with automated pipelines and standardized processes to build, deploy, run, monitor, debug, and retrain machine learning and AI models.
- Develop, maintain, and enhance frameworks for machine learning model development and deployment.
- Collaborate with ML/AI builders and application owners to determine business requirements and SLAs for API-enabled services.
- Develop, maintain, and enhance infrastructure supporting machine learning services.
- Create deployment patterns for machine learning models with CI/CD pipelines and automated testing.
- Apply AI tools throughout the engineering workflow and identify opportunities to reduce effort, simplify workflows, and provide proactive guidance.
- Adopt best practices for using AI technologies across technical development.
Requirements
- 5+ years of software engineering experience.
- Proficiency in Python, Ruby, or Java.
- Experience designing and developing infrastructure and platform services across the machine learning lifecycle, including feature stores, model development, deployment, and observability.
- Experience with at least one major cloud platform; AWS is preferred.
- Curiosity and experimentation with emerging AI frameworks, including evaluating and scaling AI use safely across teams.
- Comfort with AI-assisted development tools and staying current with emerging software development approaches.
Compensation and Benefits
- Targeted annual cash compensation is $160,000-$200,000 in Denver and $190,000-$240,000 in San Francisco and New York.
- Full-time employees receive benefits and equity (RSUs).
- Employees based in Denver, San Francisco, or New York are expected to work from the office approximately 2–3 days per week or more depending on the role.
- When working outside a Gusto office, employees must maintain a secure, reliable, and consistent internet connection.
Skills
Python, Ruby, Java, MLOps, Machine Learning, AI, AWS, Cloud Platforms, Data Pipelines, Feature Stores, CI/CD, Automated Testing, Model Deployment, Observability, Ai Frameworks
Similar jobs
ML Engineering jobs
Earnin
Mountain View, CA
Build production-grade AI agents, evaluation infrastructure, and developer tooling that make AI-assisted engineering faster, safer, and reusable across teams. The role requires software engineering experience, platform or internal developer-product experience, and hands-on expertise with LLM integration and orchestration.
Nuro
Mountain View, CABuild trustworthy infrastructure for production LLM agents, closed-loop evaluation, and autonomous research workflows. The role requires strong Python and distributed-systems experience, hands-on LLM post-training and inference knowledge, and experience operating agent systems at scale.
ClickUp
United States
Build and operate large-scale ranking and retrieval systems that power search relevance, including hybrid lexical/vector search, embeddings, query understanding, and permission-aware retrieval. Requires a bachelor's degree and 5+ years of ML engineering experience in ranking or information retrieval.
Atomicmachines
Emeryville, CA
Build and operate production ML infrastructure spanning training, deployment, serving, monitoring, data pipelines, and feedback-driven retraining. The role requires strong MLOps and DevOps experience, Python and SQL proficiency, and ownership of reliable cloud-based systems.
Tessera Labs
San Jose, CABuild and scale post-training, reinforcement-learning, evaluation, and inference systems for long-horizon agents operating over complex enterprise software. The role requires strong Python and PyTorch or JAX skills, distributed GPU experience, empirical rigor, and the ability to take research results into production.