Applied AI Inference Engineer, Baseten
$160-275k
• Offers Equity
ML application builder for data scientists
Open for applications
ML application builder for data scientists
21-100 employees
Open for applications
$160-275k
• Offers Equity
21-100 employees
Company mission
To make machine learning accessible to every organisation.
Role
Who you are
- Bachelor's, Master's, or Ph.D. degree in Computer Science, Engineering, Mathematics, or related field
- 1+ years of professional work experience in a fast-paced, high-growth environment
- Demonstrated experience with one or more general-purpose programming languages in a production-level environment, with a strong preference for Python
- Familiarity with AI/ML pipelines and the lifecycle of ML model development and deployment
- Strong communication skills, particularly on complex technical topics
- Experience in building or optimizing AI/ML projects is highly valued
What the job involves
- As an Applied AI Inference Engineer at Baseten, you will partner directly with customers to architect, build, and deploy high-scale production AI applications on Baseten’s platform
- You’ll own the journey with customers from initial exploration to production deployment, translating ambiguous business goals into reliable, observable services with clear quality, latency, and cost outcomes
- This role is a great fit for entrepreneurial engineers who want a front-row view into how modern companies adopt AI at scale and who enjoy working across product, software development, performance engineering, and customer-facing implementations
- To be clear, this is an engineering role with hands-on coding and software development that also includes aspects of product management, technical customer success, and pre-sales solution engineering mixed in
- Develop and maintain software systems and product features using one or more general-purpose programming languages in a production-level environment, with a preference for Python due to its relevance in ML projects
- Drive customer impact by designing, implementing, and deploying Baseten solutions end-to-end (problem framing → evaluation → production deployment → monitoring). This involves working with customers’ engineering teams at every stage of the customer journey including: sales, implementation, and expansion
- Deliver with velocity: turn vague objectives into clear specs and well-defined PoCs so we can rapidly ship well-tested services and outcomes for our customers
- Optimize and enhance AI/ML projects, contributing to the continuous improvement of our technical stack. This includes developing features and PRDs with other engineering and product orgs
- Own products and customer projects end-to-end, functioning as both an engineer, project manager, and product manager, with a focus on user empathy, project specification, and end-to-end execution
- Navigate ambiguity and exercise good judgment on tradeoffs and tools needed to solve problems, avoiding unnecessary complexity
- Demonstrate pride, ownership, and accountability for your work, expecting the same from your teammates
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 built 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 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, especially across engineering and marketing. Having hired 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 uncertain on how to actually use.
Simplifying the complex world of artificial intelligence for business is no mean feat but the three founders, who met whilst working as engineers at Gumroad, have a lot of experience in how to navigate this gap in the market.
Kirsty
Company Specialist at Welcome to the Jungle
Company
Funding (last 2 of 6 rounds)
Jan 2026
$300m
LATE VC
Sep 2025
$150m
SERIES D
Total funding: $585m
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 42 more jobs at Baseten