Senior Software Engineer, ML Infrastructure

SambaNova, Inc.

United States

Remote

USD 200,000 - 275,000

Full time

14 days+
Application generator

Get a reply from this employer — a resume and cover letter tailored to exactly what they’re hiring for.

Get past ATS filters

Benefits offered by this job

Health insurance
HSA contribution
Dental & Vision
Life insurance
Employee assistance program
Headspace subscription
Gympass+
Wellbeing programs

Job summary

SambaNova Systems is seeking a Senior Software Engineer, ML Infrastructure to design, build, and operate production-grade inference infrastructure that powers SambaNova’s serving stack on our RDU architecture, enabling high-throughput, low-latency services exposed through SambaStack and SambaCloud.

You will own end-to-end systems, collaborate with ML, compiler, runtime, and product teams to ship inference features from prototype to production, and contribute to design reviews, code reviews, and

Qualifications

  • Bachelor’s degree in Computer Science, Electrical Engineering, or related field.
  • 5+ years of industry experience building and operating large-scale distributed systems, ideally in ML serving.
  • Strong software engineering fundamentals: algorithms, data structures, concurrency, and systems design.
  • Experience designing and maintaining production services with strict latency, throughput, and availability requirements.
  • Working knowledge of modern LLM inference techniques and familiarity with open-source serving stacks such as vLLM, TensorRT-LLM, or SGLang.
  • Proficiency in Python.
  • Experience collaborating across teams to deliver complex, system-level engineering solutions.

Responsibilities

  • Design and productionize advanced inference techniques on RDU to optimize for performance and cost.
  • Own SambaNova’s integration with vLLM and adjacent serving frameworks, adapting them to RDU’s architecture.
  • Own the public inference API surface exposed through SambaStack and SambaCloud.
  • Build and maintain the accuracy verification and regression infrastructure that gates every inference feature shipped to customers.
  • Partner with ML, compiler, runtime, and product teams to take inference features from prototype to production.
  • Contribute to technical design discussions, code reviews, and architectural decisions as a senior individual contributor.

Skills

Python
Distributed systems
ML serving
Latency optimization
Algorithms & data structures
Concurrency

Education

Bachelor’s degree in Computer Science, Electrical Engineering, or related field

Tools

vLLM
TensorRT-LLM
SGLang
SambaStack
SambaCloud

Job description

The era of pervasive AI has arrived. In this era, organizations will use generative AI to unlock hidden value in their data, accelerate processes, reduce costs, drive efficiency and innovation to fundamentally transform their businesses and operations at scale.

SambaNova Suite™ is the first full-stack, generative AI platform, from chip to model, optimized for enterprise and government organizations. Powered by the intelligent SN40L chip, the SambaNova Suite is a fully integrated platform, delivered on-premises or in the cloud, combined with state-of-the-art open-source models that can be easily and securely fine-tuned using customer data for greater accuracy. Once adapted with customer data, customers retain model ownership in perpetuity, so they can turn generative AI into one of their most valuable assets.

Overview

The Senior Software Engineer, ML Infrastructure will be responsible for designing, building, and operating the production-grade inference infrastructure that powers SambaNova’s serving stack on our Reconfigurable Dataflow Unit (RDU) architecture. SambaNova is an inference-first company, and this role sits at the heart of that mission: turning state-of-the-art inference techniques into reliable, high-throughput, low-latency services exposed to customers through SambaStack and SambaCloud. The engineer will own end-to-end systems spanning request scheduling, advanced decoding algorithms, caching layers, API surfaces, and the accuracy infrastructure that keeps the stack trustworthy. This role partners closely with ML, compiler, runtime, and product teams to ship inference features from prototype to production.

Qualifications
  • Bachelor’s degree in Computer Science, Electrical Engineering, or related field
  • 5+ years of industry experience building and operating large-scale distributed systems, ideally in ML serving
  • Strong software engineering fundamentals: algorithms, data structures, concurrency, and systems design
  • Experience designing and maintaining production services with strict latency, throughput, and availability requirements
  • Working knowledge of modern LLM inference techniques and familiarity with open-source serving stacks such as vLLM, TensorRT-LLM, or SGLang
  • Proficiency in Python
  • Experience collaborating across teams to deliver complex, system-level engineering solutions
Key responsibilities
  • Design and productionize advanced inference techniques on RDU to optimize for performance and cost. Key areas include speculative decoding, constrained decoding, function/tool calling, prompt caching, and long-context inference.
  • Own SambaNova’s integration with vLLM and adjacent serving frameworks, adapting them to RDU’s architecture.
  • Own the public inference API surface exposed through SambaStack and SambaCloud.
  • Build and maintain the accuracy verification and regression infrastructure that gates every inference feature shipped to customers.
  • Partner with ML, compiler, runtime, and product teams to take inference features from prototype to production.
  • Contribute to technical design discussions, code reviews, and architectural decisions as a senior individual contributor.

Base Salary Range:

$200,000—$275,000 USD

EEO Policy

SambaNova Systems is an Equal Opportunity/Affirmative Action Employer. All qualified applicants will receive consideration for employment without regard basis of age (40 and over), color, disability, gender identity, genetic information, marital status, military or veteran status, national origin/ancestry, race, religion, creed, sex (including pregnancy, childbirth, breastfeeding), sexual orientation, and any other applicable status protected by federal, state, or local laws.

Benefits Summary for US-Based, Full-Time Employment Positions

SambaNova offers a competitive total rewards package, including the base salary, plus equity and benefits. We cover 95% premium coverage for employee medical insurance, and 77% premium coverage for dependents and offer a Health Savings Account (HSA) with employer contribution. We also offer Dental, Vision, Short/Long term Disability, Basic Life, Voluntary Life, and AD&D insurance plans in addition to Flexible Spending Account (FSA) options like Health Care, Limited Purpose, and Dependent Care. Our library of well-being benefits available to you and your dependents includes a full subscription to Headspace, Gympass+ membership with access to physical gyms, One Medical membership, counseling services with an Employee Assistance Program, and much more.

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Senior Software Engineer - ML Infrastructure
Senior Software Engineer - ML Infrastructure

SambaNovaSystems • United States

On-site
USD 200,000 - 275,000
Health insurance
Health Savings Account (HSA)
Headspace subscription
+2
Senior Principal Machine Learning Engineer
Senior Principal Machine Learning Engineer

SambaNovaSystems • San Jose (CA)

On-site
USD 220,000 - 300,000
Equity
Health insurance
HSA
+1
Senior Principal Machine Learning Engineer
Senior Principal Machine Learning Engineer

SambaNova • San Jose (CA)

On-site
USD 220,000 - 300,000
Health insurance
HSA
Equity
+1
Principal Cloud Platform Engineer
Principal Cloud Platform Engineer

SambaNova • San Jose (CA)

On-site
USD 144,000 - 189,000
Health insurance
HSA
Gympass+
+1
Principal Cloud Platform Engineer
Principal Cloud Platform Engineer

SambaNova • Austin (TX)

On-site
USD 144,000 - 189,000
Health insurance
HSA with employer contribution
Dental, Vision
+2
Director, Software Engineering San Jose, California, United States
Director, Software Engineering San Jose, California, United States

SambaNova • San Jose (CA)

On-site
USD 245,000 - 325,000
95% premium coverage for employee medical insurance
77% premium coverage for dependents
Health Savings Account (HSA)
+2
Software Engineering Director - Inference Platform
Software Engineering Director - Inference Platform

SambaNovaSystems • San Jose (CA)

On-site
USD 245,000 - 325,000
Health benefits (premium covered)
Headspace
Gympass+ membership
Forward Deployment Engineer
Forward Deployment Engineer

SambaNova, Inc. • United States

On-site
USD 67,000 - 83,000
Headspace
Gympass+
One Medical
+7
Cloud Platform Architect
Cloud Platform Architect

SambaNova • Austin (TX)

On-site
USD 245,000 - 325,000
Equity
Medical insurance
Headspace
+1
Director of Product, Ecosystem
Director of Product, Ecosystem

SambaNova • San Jose (CA)

On-site
USD 243,000 - 297,000
Equity
Health insurance
HSA
+4