AI Performance Engineer - New Grad

About the Team

About the Role

Responsibilities

• 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

Preferred Qualifications

• 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
• Base Salary Range:
• Base Pay Range$131,200—$196,800 USD
• Submission Guidelines Please 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 Policy SambaNova 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 Positions SambaNova 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.
• Submission Guidelines
• Please 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 Policy
• SambaNova 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 Positions
• SambaNova 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.