Software Engineer - Testing Frameworks

Baseten; Montreal (administrative region); On-site; CAD 230,000 - 460,000; Full time

Benefits offered by this job

Competitive compensation & equity

Comprehensive medical/dental/vision

Flexible PTO

Parental leave

401(k)

Exposure to ML startups

Job summary

Baseten is seeking talented and experienced Software Engineers to join our Platform team within the Infrastructure organization. You will own the systems that let every engineer prove their code works before it reaches production, with a focus on fast, reliable testing across unit, integration, and load scenarios.

Your work includes building frameworks, harnesses, and feedback loops to define what well-tested means at Baseten, while ensuring resilience and performance for mission-critical AI

Qualifications

• Strong proficiency in Go and/or Python with hands-on testing tooling experience.
• Deep knowledge of testing frameworks and their internals, and ability to extend them.
• Experience designing integration test infrastructure for distributed systems.
• Experience with load and performance testing tools and interpreting results.

Responsibilities

• Own Baseten's testing strategy end to end, defining standards and tooling.
• Build and maintain unit, integration, load and performance testing frameworks.
• Design end-to-end test infra provisioning realistic dependencies on demand.
• Build resilience and chaos testing capabilities to ensure graceful degradation.
• Improve test reliability by addressing flakes and CI efficiency.
• Incorporate observability into testing with dashboards and signals.
• Mentor engineers and raise testing practice across the org.

Skills

Go

Python

Testing frameworks

Distributed systems

Tools

pytest

testcontainers

K6

Locust

JMeter

Gatling

Job description

About Baseten

Baseten powers mission-critical inference for the world's most dynamic AI companies, like Cursor, Notion, OpenEvidence, Abridge, Clay, Gamma and Writer. By uniting applied AI research, flexible infrastructure, and seamless developer tooling, we enable companies operating at the frontier of AI to bring cutting-edge models into production. We're growing quickly and recently raised our $1.5B Series F, led by Altimeter Capital, Conviction Partners, and Spark Capital. Join us and help build the platform engineers turn to to ship AI products.

Baseten powers mission-critical inference for the world's most dynamic AI companies, like Cursor, Notion, OpenEvidence, Abridge, Clay, Gamma and Writer. By uniting applied AI research, flexible infrastructure, and seamless developer tooling, we enable companies operating at the frontier of AI to bring cutting-edge models into production. We're growing quickly and recently raised our $1.5B Series F, led by Altimeter Capital, Conviction Partners, and Spark Capital. Join us and help build the platform engineers turn to to ship AI products.

THE ROLE

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.

Responsibilities

• 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 — ephemerical 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.

Requirements

• 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 ephemerical 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.

Benefits

• Competitive compensation, including meaningful equity.
• 100% coverage of medical, dental, and vision insurance for employee and dependents
• Flexible PTO policy including company wide Winter Break (our offices are closed from Christmas Eve to New Year's Day!)
• Paid parental leave
• Fertility and family‑building stipend through Carrot
• Company‑facilitated 401(k)
• Exposure to a variety of ML startups, offering unparalleled learning and networking opportunities.

At Baseten, we are committed to fostering a diverse and inclusive workplace. We provide equal employment opportunities to all employees and applicants without regard to race, color, religion, gender, sexual orientation, gender identity or expression, national origin, age, genetic information, disability, or veteran status. We are an Equal Opportunity Employer and will consider qualified applicants with criminal histories in a manner consistent with applicable law (by example, the requirements of the San Francisco Fair Chance Ordinance, where applicable).

Compensation Range: $165K - $330K