Compiler Engineer - New Grad
$131k - $197k • Austin, Texas, United States; San Jose, California, United States
Posted 2d ago
About the job
SambaNova is seeking a new grad Compiler Engineer to work on core areas of our compiler, which powers the RDU (Reconfigurable Dataflow Unit) chip. This role involves working on the compiler's IR and pass infrastructure, model lowering and partitioning, tensor tiling, memory management, and mapping operations onto the accelerator fabric. You will gain insight into how compiler decisions impact chip performance, write production code in C++ and Python, perform low-level debugging, and collaborate with runtime and hardware teams. Senior engineers will provide code reviews and mentorship.
Responsibilities
- 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.
Requirements
- 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.
Benefits
- Equity
- 95% premium coverage for employee medical insurance
- 77% premium coverage for dependents
- Health Savings Account (HSA) with employer contribution
- Dental insurance
- Vision insurance
- Short-term disability insurance
- Long-term disability insurance
- Basic life insurance
- Voluntary life insurance
- Accidental Death & Dismemberment (AD&D) insurance
- Flexible Spending Account (FSA) options (Health Care, Limited Purpose, Dependent Care)
- Headspace subscription
- Gympass+ membership
- One Medical membership
- Counseling services with an Employee Assistance Program