Sambanovasystems

Compiler Engineer - New Grad

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

The role

You will work on core areas of the compiler including IR and pass infrastructure, model lowering, tensor tiling, and memory management. You will write production code in C++ and Python to power the RDU.

What you will do

  • Implement and optimize passes in the compiler's IR and pass infrastructure
  • Lower and partition machine learning models for execution on the RDU
  • Develop tensor tiling and memory management for on-chip memory
  • Map operations onto the RDU fabric to improve model performance
  • Debug compiler and performance issues at a low level in C++ and Python

Required qualifications

  • B.S. in Computer Science, Electrical Engineering, Computer Engineering, or related field
  • Experience in machine learning systems development, including ML compilers, LLM inference kernels, or kernel generation
  • Strong programming skills in Python and C/C++
  • Experience programming GPUs or other AI accelerators

Nice to have

  • M.S. or Ph.D. in Computer Science, Electrical Engineering, Computer Engineering, or related field
  • Experience with ML compiler frameworks such as MLIR or StableHLO
  • Experience with LLM kernel libraries
  • Machine learning systems research experience
  • Open source project ownership or contributions
  • Experience building AI agent driven tools or automation

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 technology is centered on the RDU (Reconfigurable Dataflow Unit).

Compiler Engineer - New Grad at Sambanovasystems | JobZoey