AI Engineer, Baseten
$175-190k
+ Equity
ML application builder for data scientists
Be an early applicant
ML application builder for data scientists
101-200 employees
Be an early applicant
$175-190k
+ Equity
101-200 employees
Company mission
To make machine learning accessible to every organisation.
Role
Who you are
- 3+ years of experience in AI/automation engineering, workflow automation, or a technical operations role, ideally at a high-growth, AI-native infrastructure or B2B company
- You've shipped production agents on Vercel. Build custom internal apps and agents on Vercel, with durable workflows and sandboxed execution, so what you ship runs reliably in production instead of as one-off scripts
- Genuine fluency with AI coding assistants and agent tooling (Claude Code, Cursor, Codex, or similar)
- Direct experience with a CRM or system-of-record platform (Salesforce or similar)
- A track record of owning end-to-end AI and automation workflows
- Proficiency with integration and automation platforms (n8n, Zapier, Make, Workato, or similar) and a fundamental understanding of APIs and webhooks
- Experience in high-growth technology companies, ideally in infrastructure, cloud, or operations-heavy environments
Desirable
- Familiarity with GPU infrastructure, capacity planning, or fleet management concepts
- Comfort building reporting and alerting on a data warehouse (BigQuery, Databricks) and BI layer (Sigma, Hex) — not as a data engineer, but as a consumer and builder on top of the data layer
- Experience building with agent platforms (Gumloop, n8n, Notion Agents, etc.) and/or agent frameworks (Vercel AI SDK, Claude Agent SDK, Mastra, etc.)
- Experience with high-stakes, operationally complex environments where mistakes have real business impact
- Challenging work and exposure to the operational core of a fast-scaling AI infrastructure company
What the job involves
- Ship AI-powered workflows for Compute and C3: build the agents and automations that give capacity analysts, ops leads, and engineers real leverage, off-loading manual and repetitive work like data pipeline cleanup, migration troubleshooting, and capacity investigation
- Get insights in front of the team: turn fleet utilization, allocation, and demand-forecasting data into the dashboards, alerts, and recommendations that C3 and Compute leadership actually act on. A build isn't done until the team is using it
- Audit the stack and generate your own backlog: Compute runs on a mix of tooling with real overlap and real gaps. Walk in with a point of view, identify what's missing or broken, and prioritize without waiting to be handed a roadmap
- Ship team-productivity workflows fast: triage asks from Supply, Demand, and C3, find the low-hanging fruit, and build it. A good week looks like an ops lead asking for something Monday and having it live by Wednesday
- Know when to go custom: not everything belongs in a point-and-click tool. Spin up bespoke AI-powered solutions in Claude Code when the problem calls for it, and make that call with judgment, not default
- Think in systems, not solutions: every workflow you build has upstream and downstream implications across Supply, Demand, and Engineering. Anticipate them, design for them, and don't create technical debt someone else has to unwind six months later
- Integrate third-party APIs and internal systems to sync events and information across disparate tools: ensuring data flows reliably between C3, the CRM, and the systems Compute depends on every day
- Document what you build: if people can't find it, understand it, or trust it, it doesn't matter how well it works
Our take
As with its mathematical namesake, Baseten focuses on simplifying the complex. The San Francisco startup focuses on simplifying the integration of artificial intelligence into businesses. The tools that Baseten build demolish the need for businesses to hire specialised engineers, utilising the huge strides in artificial intelligence that we’ve seen in the last few years, all without adding workload to businesses.
With serious funding behind it, Baseten has grown its team by three times year-over-year over the last five years. Specifically, it's hired across engineering and marketing, gaining talent from companies such as GitHub, Google, Uber, Amazon, and Palantir. Part of Baseten's focus is on helping companies implement the tools that they have at their disposal, a long-overlooked aspect of machine learning that many businesses have been eager to champion, but are uncertain about how to actually use.
Simplifying the complex world of artificial intelligence for business is no mean feat, but with hundreds of millions in funding behind it and a $5 billion valuation reported in 2026, the company is well on its way to market domination. What's more, the three founders, who met whilst working as engineers at Gumroad, have a lot of experience in navigating this gap in the market.
Kirsty
Company Specialist at Welcome to the Jungle
Company
Funding (last 2 of 7 rounds)
Jun 2026
$1.5bn
SERIES F
Jan 2026
$300m
SERIES E
Total funding: $2.1bn
Company benefits
- Remote-first work environment. The Baseten team is welcome to work from wherever they want; fully remote, in our San Francisco office, or a mix of both. Today, our team (including our founding team) is spread across the United States, Canada, and Armenia. We provide a $1,000 stipend for you to make your home-office comfortable and productive
- Regular in-person team summits. We get together as a team three times a year to plan, workshop, and most importantly, get to know each other better
- Unlimited PTO. We ask that everyone take at least 4 weeks of vacation. And we have a company-wide break between Christmas and New Year's Day
- Full healthcare coverage. Medical, dental and vision insurance for you and your family
- Paid parental leave. 16-weeks fully paid parental leave (adoptive and non-birth parents included) and flexibility with schedules while returning to work
- Company-sponsored 401(k) for you to contribute to
- Learning and development budget. We encourage you to take classes, attend conferences, and invest in your craft and we’ll cover expenses to make it happen
Company HQ
Union Square, San Francisco, CA
Leadership
Tuhin Srivastava
(Co-Founder)Previously Co-Founder at Shape, Machine Learning at Gumroad and Skulpt.
Amir Haghighat
(Co-Founder)Previously Head of Engineering at Gumroad, Software Engineer at Yelp and Engineering Manager at Clover Health.
Phillip Howes
(Co-Founder)Previously Co-Founder at Shape, Machine Learning Engineer at Gumroad and Research Assistant at the University of Sydney.
Salary benchmarks
We don't have enough data yet to provide salary benchmarks for this role.
Submit your salary to help other candidates with crowdsourced salary estimates.
Share this job
View 85 more jobs at Baseten