Overview
Company
Profile match
Impact
Conditions
Benefits
Hiring process
Similar jobs

Baseten

Baseten is an AI infrastructure platform designed to help developers and machine learning teams deploy, serve, and scale open-source and custom AI models. The platform provides performant, low-latency inference infrastructure alongside developer tools like Truss, an open-source model packaging framework. Headquartered in San Francisco, California, Baseten enables companies to run state-of-the-art models in production seamlessly without managing underlying cloud infrastructure.

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.

THE ROLE

Baseten is building its own GPU infrastructure for large-scale inference. As we move into large scale, high-density NVIDIA systems, the hardest failures are intermittent, cross-layer, and difficult to prove: RoCE congestion, InfiniBand stalls, ECN/DCQCN mis-tuning, bad optics, RNIC issues, host kernel stalls, GPU driver problems, and workload symptoms that look like network problems, but are not.

We are hiring a Lead Software Engineer to build a first-class observability and root-cause analysis system for GPU fabrics. This is a hard distributed systems problem, not a dashboarding problem. The system will collect high-volume signals from switches, hosts, active probes, and inference services; reduce and correlate them in real time; understand topology and service ownership; and produce actionable diagnosis while an incident is still unfolding.

This role sits at the boundary between networking and inference software. RDMA data paths, GPUDirect transfers, prefill/decode disaggregation, KV cache movement, request routing, and workload backpressure can all create fabric symptoms or hide real fabric failures. The goal is to tell an operator, quickly and with evidence, whether an incident is caused by the fabric, host, NIC, GPU, RDMA path, scheduler, or serving layer - and what to do next.

EXAMPLE INITIATIVES

  • Real-time telemetry engine - Build the ingestion, reduction, storage, and query path for high-cardinality fabric, host, GPU, and workload telemetry.

  • Service-aware fabric diagnosis - Build collectors, probes, and topology-aware correlation to detect latency, drops, stalls, congestion, bad paths, and degradation.

  • Software-aware RDMA diagnosis - Tie network behavior to RDMA operations, GPUDirect paths, KV cache transfers, prefill/decode disaggregation, and request latency.

RESPONSIBILITIES

  • Own Baseten’s GPU fabric observability and root-cause analysis architecture.

  • Build telemetry pipelines across switches, NICs, hosts, GPUs, Kubernetes, and inference services.

  • Model topology, flow paths, service ownership, and failure domains.

  • Separate true fabric faults from host, NIC, GPU, kernel, driver, RDMA, scheduler, and workload failures.

  • Create clear operator workflows for triage, remediation, and post-incident learning.

REQUIREMENTS

  • Staff-level or senior staff-level experience building production infrastructure software.

  • Strong distributed systems background, especially streaming systems, telemetry pipelines, diagnostics, or control-plane software.

  • Experience building systems that process high-volume, high-cardinality, noisy operational data.

  • Understanding of networking fundamentals and high-performance networks

  • Ability to work with low-level infrastructure signals and build practical correlation, anomaly detection, or root-cause analysis systems.

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.

Apply now to embark on a rewarding journey in shaping the future of AI! If you are a motivated individual with a passion for machine learning and a desire to be part of a collaborative and forward-thinking team, we would love to hear from you.

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).

Recommended for you based on this role

Similar stack
Same company
In your city
$165k – $330k per year • Remote • Full-Time • 4+ year exp • San Francisco
Go
Python
AI/ML
Claude
Claude Code
Cursor
LLM
Spark
DevOps
CI/CD
Docker
Kubernetes
Management
Notion
Apply
$165k – $330k per year • Remote • Full-Time • San Francisco
Go
Python
Java
Java
Testcontainers
AI/ML
Cursor
Spark
DevOps
CI/CD
Docker
Kubernetes
Management
Notion
QA
Gatling
JMeter
k6
Locust
Pytest
Apply
$165k – $330k per year • Remote • Full-Time • San Francisco
Go
Python
AI/ML
Cursor
Spark
DevOps
ArgoCD
CI/CD
GitOps
Kubernetes
Progressive Delivery
Apply
$165k – $330k per year • Remote • Full-Time • San Francisco
Go
Python
Rust
Databases
ClickHouse
AI/ML
Anomaly Detection
Cursor
Spark
DevOps
Grafana
OpenTelemetry
Prometheus
SLI/SLO/SLA
Apply
Head of Environment 9 days ago
$200k – $240k per year • In office • Full-Time • 10+ year exp • San Francisco
AI/ML
Cursor
Apply
$165k – $330k per year • Remote • Full-Time • 2+ year exp • Bachelor's Degree • San Francisco
Python
AI/ML
ComfyUI
Cursor
Spark
Whisper
DevOps
Docker
Management
Notion
Apply
Solutions Architect 9 days ago
Remote/Hybrid • Bachelor's Degree • San Francisco
AI/ML
Cursor
Embeddings
SGLang
Management
Notion
Apply
$125k – $170k per year • Remote • Full-Time • 4+ year exp • New York
AI/ML
Cursor
Cybersecurity
Okta
Management
Google Workspace
Notion
Apply
Full-Time • 8+ year exp • Bachelor's Degree • San Francisco
Python
AI/ML
Cursor
Fine-tuning
Knowledge Distillation
Quantization
Spark
Management
Notion
Apply
$180k – $330k per year • Remote • Full-Time • 8+ year exp • San Francisco
AI/ML
Cursor
Cybersecurity
HIPAA
SOC 2
Management
Notion
Apply
Career impact
Discover how this job can transform your career
Get a personal career forecast for this job - salary uplift, next-level role, skill boost and a 3-year financial impact, all calculated from your profile.
Personal salary uplift vs. your current pay
Your 3-year career trajectory
Skills you will level up in this role
3-year financial impact in dollars
Create free account
Free forever • Less than a minute • No credit card

Work setup

Location
San Francisco
Remote work
Remote
Employment
Full-Time

Compensation

Salary
$200k – $380k per year
Benefits
Parental leave, Vision insurance
Equity
Equity stake in a tech company