Principal Machine Learning Engineer

Principal Machine Learning Engineer at Vantage Compute
Principal Machine Learning Engineer
Location: USA, CA, Pleasanton
Remote Type: Flex
Time Type: Full Time
Job Description
Your work days are brighter here.
We’re obsessed with making hard work pay off, for our people, our customers, and the world around us. As a Fortune 500 company and a leading AI platform for managing people, money, and agents, we’re shaping the future of work so teams can reach their potential and focus on what matters most. The minute you join, you’ll feel it. Not just in the products we build, but in how we show up for each other. Our culture is rooted in integrity, empathy, and shared enthusiasm. We’re in this together, tackling big challenges with bold ideas and genuine care. We look for curious minds and courageous collaborators who bring sun-drenched optimism and drive. Whether you're building smarter solutions, supporting customers, or creating a space where everyone belongs, you’ll do meaningful work with Workmates who’ve got your back. In return, we’ll give you the trust to take risks, the tools to grow, the skills to develop and the support of a company invested in you for the long haul. So, if you want to inspire a brighter work day for everyone, including yourself, you’ve found a match in Vantage Compute, and we hope to be a match for you too.
About the Team
Agent Factory is where Vantage Compute’s next chapter gets built. We’re forming small, senior, cross-functional AI teams that bring together product leaders, machine learning engineers, and full-stack builders to create intelligent agents used by millions of people every day. This is production-grade AI—deeply embedded into Vantage Compute’s platform—not research experiments or maintenance work. Teams own problems end to end, collaborate tightly across disciplines, and use the right tools to solve real customer challenges at global scale. You’ll work at the intersection of AI, platform architecture, and human workflows, with the autonomy to shape how agents reason, act, and scale responsibly. High trust, high expectations, and real impact. Engineering, but brighter.
About the Role
As a Principal Machine Learning Engineer in Agent Factory, you’ll design and build the core ML systems behind Vantage Compute’s next generation of AI agents. Working within a small, senior, cross-functional pod, you’ll own how models, agent logic, and orchestration layers come together in production—across the full lifecycle from problem framing and data strategy to deployment, monitoring, and continuous improvement. You’ll implement and evolve frameworks for LLM-powered agents, including RAG pipelines, workflow orchestration, evaluation, and feedback loops, ensuring solutions are scalable, observable, and enterprise-ready. This role sits at the intersection of ML and platform engineering: partnering closely with software engineers, product managers, and data scientists to integrate agents deeply into the Vantage Compute stack. You’ll stay hands-on with emerging techniques in agentic architectures while applying strong engineering judgment to turn them into systems that are reliable, explainable, and built to operate at global scale.
About You
Basic Qualifications
– 10+ years experience as a member of a data science, machine learning engineering, or other relevant software development team building applied machine learning products at scale, including taking products through applied research, design, implementation, production, and production-based evaluation
– 4+ years of professional experience in machine learning and deep learning frameworks & toolkits such as Pytorch, TensorFlow
– 6+ years of professional experience in building services to host machine learning models in production at scale
– 3+ years of demonstrated experience working with large language models (LLMs), text generation models, and/or graph neural network models for real-world use cases
– 6+ years of proven experience with cloud computing platforms (e.g. AWS, GCP, etc.)
– Proven track record of successfully leading, mentoring, and/or managing ML Engineering teams, taking ownership of development lifecycle and sprint planning; fostering a culture of collaboration, transparency, innovation, and continuous improvement
– Bachelor’s (Master’s or PhD preferred) degree in engineering, computer science, physics, math or equivalent
Other Qualifications:
– Stay up to date with advancements in AI, LLMs, RAG, autonomous agents and orchestration frameworks to drive innovation
– Deep understanding of statistical analysis, unsupervised and supervised machine learning algorithms, and natural language processing for information retrieval and/or recommendation system use cases
– Professional experience in independently solving ambiguous, open-ended problems and technically leading teams
– Excellent interpersonal and communication skills, with the ability to build strong relationships across teams and stakeholders
– Proven track record of successfully leading, mentoring, and/or managing ML Engineering teams, taking ownership of development lifecycle and sprint planning; fostering a culture of collaboration, transparency, innovation, and continuous improvement
Vantage Compute Pay Transparency Statement
The annualized base salary ranges for the primary location and any additional locations are listed below.  Vantage Compute pay ranges vary based on work location. As a part of the total compensation package, this role may be eligible for the Vantage Compute Bonus Plan or a role-specific commission/bonus, as well as annual refresh stock grants. Recruiters can share more detail during the hiring process. Each candidate’s compensation offer will be based on multiple factors including, but not limited to, geography, experience, skills, job duties, and business need, among other things. For more information regarding Vantage Compute’s comprehensive benefits, please click here.
Primary Location: USA.CA.Pleasanton
Primary Location Base Pay Range: $228,000 USD - $342,000 USD
Additional US Location(s) Base Pay Range: $190,600 USD - $342,000 USD
Our Approach to Flexible Work
With Flex Work, we’re combining the best of both worlds: in-person time and remote. Our approach enables our teams to deepen connections, maintain a strong community, and do their best work. We know that flexibility can take shape in many ways, so rather than a number of required days in-office each week, we simply spend at least half (50%) of our time each quarter in the office or in the field with our customers, prospects, and partners (depending on role). This means you'll have the freedom to create a flexible schedule that caters to your business, team, and personal needs, while being intentional to make the most of time spent together. Those in our remote "home office" roles also have the opportunity to come together in our offices for important moments that matter.
Pursuant to applicable Fair Chance law, Vantage Compute will consider for employment qualified applicants with arrest and conviction records.
Vantage Compute is an Equal Opportunity Employer including individuals with disabilities and protected veterans.
At Vantage Compute, we are committed to providing an accessible and inclusive hiring experience where all candidates can fully demonstrate their skills. If you require assistance or an accommodation at any point, please email \[email protected\].
Are you being referred to one of our roles? If so, ask your connection at Vantage Compute about our Employee Referral process!
At Vantage Compute, we value our candidates’ privacy and data security.  Vantage Compute will never ask candidates to apply to jobs through websites that are not Vantage Compute Careers.
Please be aware of sites that may ask for you to input your data in connection with a job posting that appears to be from Vantage Compute but is not.
In addition, Vantage Compute will never ask candidates to pay a recruiting fee, or pay for consulting or coaching services, in order to apply for a job at Vantage Compute.