Sambanovasystems

Senior Principal Model Bring-up Engineer

Austin, Texas, United States; San Jose, California, United States. Posted 16 days ago

The role

You will bring new open-weight LLM architectures up on SambaNova's RDU dataflow accelerators and into production serving. This involves adapting models, mapping them across chips and hosts, compiling, and validating them numerically. You will also build shared inference features like continuous batching and speculative decoding.

What you will do

  • Bring up new LLM architectures end to end
  • Perform model adaptation, checkpoint conversion, and multi-chip/multi-host mapping
  • Compile and perform strict numerical validation
  • Handle production serving with latency and throughput numbers
  • Build inference features as shared modules including continuous batching, prompt caching, and speculative decoding
  • Implement FP8 / FP4, tensor and expert parallelism
  • Debug where model, compiler and hardware meet

Required qualifications

  • B.S. in Computer Science, Electrical Engineering, or related field
  • 8+ years of industry experience in ML systems, inference, or compiler engineering
  • Strong Python and PyTorch internals experience
  • Some C++
  • Deep transformer-inference knowledge, including KV caching, GQA and sliding window attention, MoE routing, prefill vs decode, batching, and speculative decoding
  • Hands-on experience with accelerator bring-up, ML compilers (MLIR, XLA, TVM, Triton, torch-mlir), or inference engines (vLLM, SGLang, TensorRT-LLM)
  • Rigor in numerical debugging
  • Comfort with Bazel, Slurm, and large monorepos
  • Clear writing and evidence-backed claims

Nice to have

  • M.S. or Ph.D. in Computer Science, Electrical Engineering, or related field
  • Experience with dataflow or spatial compilers
  • Experience with kernel DSLs
  • Experience with FP8 and FP4 inference
  • Experience with Mamba and SSMs
  • Experience with expert parallelism
  • Experience with multi-agent AI engineering workflows
  • Open-source inference or compiler contributions

Benefits

  • Base salary plus equity
  • 95% premium coverage for employee medical insurance
  • 77% premium coverage for dependents
  • Health Savings Account (HSA) with employer contribution
  • Dental, Vision, Short/Long term Disability, Basic Life, Voluntary Life, and AD&D insurance plans
  • Flexible Spending Account (FSA) options like Health Care, Limited Purpose, and Dependent Care
  • Headspace subscription
  • Gympass+ membership
  • One Medical membership
  • Counseling services with an Employee Assistance Program

Before you apply

  • You must submit an application form for each position for which you believe you are qualified

About the company

SambaNova is a leader in next-generation AI infrastructure, delivering a full-stack inference platform for customers worldwide. Its core technology is the RDU (Reconfigurable Dataflow Unit) built on a dataflow architecture.

Senior Principal Model Bring-up Engineer at Sambanovasystems | JobZoey