Software Engineer, Baseten
Testing Frameworks
$165-330k
+ equity
ML application builder for data scientists
Be an early applicant
ML application builder for data scientists
101-200 employees
Be an early applicant
$165-330k
+ equity
101-200 employees
Company mission
To make machine learning accessible to every organisation.
Role
Who you are
- Strong proficiency in Go and/or Python, with hands-on experience building test tooling and libraries used by other engineers
- Deep experience with testing frameworks and their internals — pytest, Go's testing package, testify, testcontainers, or equivalents — including the ability to extend them, not just use them
- Experience designing integration test infrastructure for distributed systems, including ephemeral environment provisioning and dependency management
- Practical experience with load and performance testing tools (k6, Locust, Vegeta, JMeter, Gatling, or similar) and the judgment to interpret results and drive action from them
- Solid Kubernetes and Docker fundamentals — you can build test infrastructure that runs on and tests against real cluster environments
- Advanced understanding of CI/CD, including how to keep large test suites fast, deterministic, and trustworthy at scale
- A track record of measurably improving testing culture on an engineering team — you have opinions on what to test, what not to test, and where the ROI actually is
- Are excited about building foundational infrastructure and are comfortable working independently on ambiguous, high-impact technical challenges
What the job involves
- Baseten is seeking talented and experienced Software Engineers to join our Platform team within the Infrastructure organization
- As a senior member of Baseten's Platform Team, you will own the systems that let every engineer at Baseten prove their code works before it reaches production. Our product runs mission-critical AI inference for customers who measure downtime in dollars per second, which means our internal bar for correctness, performance, and failure tolerance has to be exceptional
- Your focus is the full testing stack: fast and reliable unit test tooling, integration harnesses that spin up realistic environments on demand, load and performance testing for GPU-backed inference workloads, and resilience testing that deliberately breaks things so our customers never have to find out what happens when a node dies mid-request
- This is a builder role with org-wide leverage. You won't be writing tests for other teams — you'll be building the frameworks, harnesses, and feedback loops that make writing good tests the path of least resistance, and you'll set the standards for what "well-tested" means at Baseten
- Own Baseten's testing strategy end to end — define the standards, the tiers, and the tooling that engineering teams build against
- Build and maintain unit, integration, load and performance testing frameworks
- Design end to end test infrastructure that provisions realistic dependencies on demand — ephemeral Kubernetes namespaces, containerized service dependencies, etc
- Build resilience and chaos testing capabilities: fault injection, network partition and latency simulation, pod and node failure scenarios, dependency degradation, and automated verification that our systems degrade gracefully and recover
- Attack test suite reliability head-on — flake detection and quarantine, test impact analysis and selective execution, parallelization, and aggressive reduction of end-to-end CI wall time
- Build observability into the testing layer itself: coverage reporting, test health dashboards, failure triage tooling, and clear signals about which areas of the codebase are under-tested
- Create project templates and shared libraries so new services start with a complete testing setup on day one
- Partner directly with product and infrastructure teams — embed where needed, understand their hardest testing problems, and turn one-off solutions into platform capabilities
- Mentor engineers across the org on testing practice and raise the collective bar through code review, documentation, and internal advocacy
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 63 more jobs at Baseten