Sambanovasystems
AI Performance Engineer - New Grad
Austin, Texas, United States; San Jose, California, United States. Posted 25 days ago
The role
You will bring up new open source models on the SambaNova stack and work to make them run faster. You will profile model execution, find bottlenecks, and apply optimizations across the compiler, runtime, and hardware layers.
What you will do
- Bring up new foundation models on the SambaNova platform using our software stack
- Profile model performance across the compiler, runtime, and hardware layers
- Identify bottlenecks and propose dataflow or scheduling optimizations for single-node and distributed systems
- Apply techniques from recent research in quantization, scheduling, and memory optimization
- Benchmark throughput and latency and track performance changes across releases
- Work with the machine learning, compiler, runtime, and hardware teams to ship performance improvements into SambaNova products
Required qualifications
- B.S. in Computer Science, Electrical Engineering, or related field
- Strong programming skills in Python or C++
- Understanding of algorithms, data structures, and numerical computing
- Experience with PyTorch, TensorFlow, or JAX
- Experience with inference frameworks such as vLLM, TensorRT, or SGLang
- GPU programming experience with CUDA or Triton
Nice to have
- M.S. or Ph.D. in Computer Science, Electrical Engineering, or related field
- Experience with LLM training or inference
- Familiarity with quantization, graph optimization, kernel fusion, or model partitioning
- Knowledge of memory hierarchy, caching, and scheduling for large model execution
- Internship or research experience in ML systems or performance optimization
- Publications or open source contributions in ML systems
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
- Full subscription to Headspace
- Gympass+ membership with access to physical gyms
- One Medical membership
- Counseling services with an Employee Assistance Program
About the company
SambaNova is a leader in next-generation AI infrastructure, delivering a full-stack inference platform for customers worldwide.