Artificial Intelligence

ML Infrastructure Engineer - New Grad

Austin, TXOnsiteInternship$131,200–$196,800 /yrPosted 11 days agoVisa sponsorship available

Is this role right for you?

Upload your resume and get a skill-by-skill breakdown — see exactly where you match, where you're close, and what to highlight. Not a mystery percentage.

Get a tailored resume highlighting what this role needs.

Required skills

Python
algorithms
data structures
concurrency
systems design
distributed systems
machine learning
LLM serving frameworks
vLLM
SGLang

Nice to have

M.S. or Ph.D
APIs
backend services
disaggregated systems
prefill/decode
open source project ownership

Role summary

SambaNova Systems is seeking a New Grad ML Infrastructure Engineer to build and operate the inference stack for their RDU accelerators. This role involves writing production Python code for inference serving, integrating with frameworks like vLLM, and implementing features such as speculative decoding and prompt caching. Responsibilities include adapting serving frameworks to the RDU architecture, building public APIs, and developing accuracy and regression tests to ensure quality. The ideal candidate will have strong programming skills in Python, understanding of algorithms, data structures, concurrency, and systems design, with coursework or project experience in distributed systems or machine learning.

SambaNova is a leader in next-generation AI infrastructure, delivering a full-stack inference platform for customers worldwide. At the core of SambaNova's technology is the RDU (Reconfigurable Dataflow Unit) — a chip built on a dataflow architecture rather than the traditional GPU model. Its decode performance is especially strong for agentic workloads like multi-turn agents, code generation, and long-running applications. RDUs are packaged into SambaRack, rack-scale hardware that lets customers deploy state-of-the-art models with better performance, greater energy efficiency, and faster time to value.

About the Team

The ML Infrastructure team builds and operates the inference stack that serves SambaNova's models on our RDU accelerators. We cover request scheduling, caching, decoding techniques, public APIs, and the accuracy testing that every inference feature must pass before it ships.

About the Role

As a new grad ML Infrastructure Engineer, you'll write production Python for our inference serving stack, including our vLLM integration and public APIs. You'll help bring features like speculative decoding, prompt caching, and tool calling into production on the RDU, and add accuracy and regression tests that catch problems before customers do.

Responsibilities

Implement inference features on the RDU such as speculative decoding, constrained decoding, tool calling, prompt caching, and long-context inference

Adapt vLLM and related serving frameworks to the RDU architecture

Build and maintain endpoints in the public inference API

Write accuracy verification and regression tests for inference features

Debug latency, throughput, and reliability issues in production services

Required Qualifications

B.S. in Computer Science, Electrical Engineering, or related field

Strong programming skills in Python

Understanding of algorithms, data structures, concurrency, and systems design

Coursework or project experience in distributed systems or machine learning

Experience with LLM serving frameworks such as vLLM, SGLang, or TensorRT-LLM

Experience with inference techniques such as speculative decoding, KV caching, or continuous batching

Experience with PyTorch

Preferred Qualifications

M.S. or Ph.D. in Computer Science, Electrical Engineering, or related field

Experience building or maintaining APIs or backend services

Internship or research experience in ML systems or backend infrastructure

Experience with disaggregated systems - prefill/decode

Open source project ownership or contributions

Base Salary Range:

Base Pay Range$131,200—$196,800 USD

Submission GuidelinesPlease note that in order to be considered an applicant for any position at SambaNova Systems, you must submit an application form for each position for which you believe you are qualified.

EEO PolicySambaNova 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 PositionsSambaNova 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.

Ready to apply?

You'll be redirected to SambaNova Systems's application page.