Artificial Intelligence Engineer, Baseten
$220-260k
+ Equity
ML application builder for data scientists
Be an early applicant
ML application builder for data scientists
101-200 employees
Be an early applicant
$220-260k
+ Equity
101-200 employees
Company mission
To make machine learning accessible to every organisation.
Role
Who you are
- 5+ years of experience building and shipping software applications
- Demonstrated experience building AI or LLM-powered products, agents, or agentic workflows that real users depend on
- Strong software engineering fundamentals and the ability to clear a real technical bar, not just prompt well
- Ability to build accurate mental models of how systems work under the hood, including the models and harnesses you're building on
- Proficiency in Python, with fluency in at least one other language
- Comfort working autonomously in a fast-moving environment with limited structure
- Ability to move between customer-facing product work and internal tooling and automation
- Strong communication skills, with the ability to bridge technical depth and business needs
Desirable
- Experience as a founding engineer or early employee at a startup
- Experience building evals, agent observability, or tooling for non-deterministic systems
- Familiarity with agent frameworks and harnesses (LangChain, Claude Code, Codex, OpenCode, MCP)
- Experience with model development methods like supervised fine-tuning, reinforcement learning, synthetic data generation, LoRA, and full fine-tunes
- Frontend fluency
What the job involves
- Are you the person on your team who builds the agent everyone else ends up using? We're looking for an AI Engineer to join our Training Product team and do that at Baseten.
- You'll build AI-driven product features for the customers training and post-training frontier models on our platform, and you'll raise the ceiling on how Baseten itself uses AI internally, turning manual workflows into agentic ones that make every other team faster
- You'll work directly with our research engineers to scope and build products, taking ideas from a research loop that already works internally to something customers can run themselves.
- This is a hands-on role with real autonomy. You'll pick the problems worth solving, build the harnesses, execution flows, and guardrails that make AI systems reliable, and own the results. If you've been shipping agents and want that to be the job, let's talk
- Build and ship agentic product experiences, including chat-style and assistant-like interfaces, from prototype to GA
- Design the harnesses, execution flows, and guardrails that make AI systems reliable in production
- Build internal automation and AI tooling that measurably increases the velocity of engineering, research, and go-to-market teams
- Partner with research engineers to scope product opportunities out of internal research workflows and turn them into customer-facing features
- Define and instrument evals so you know whether a change actually improved output quality
- Work throughout the stack (API layer, backend, agent orchestration, frontend) to implement features end to end
- Use Baseten's own training and inference products yourself to develop intuition around customer workflows
- Identify where AI can replace manual process across the company and build the thing rather than write the proposal
- Fix bugs and resolve customer issues with urgency
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 75 more jobs at Baseten