AI Performance Engineer - New Grad

Posted 5 Days Ago
2 Locations
In-Office
131K-197K Annually
Entry level
Artificial Intelligence • Hardware • Machine Learning • Natural Language Processing • Software • Generative AI
SambaNova is the #1 platform for business AI.
The Role
Brings up and optimizes foundation models on SambaNova’s AI platform. Profiles performance across compiler, runtime, and hardware layers; identifies bottlenecks; applies quantization, kernel fusion, scheduling, and memory optimizations; benchmarks inference throughput and latency; and collaborates with machine learning, compiler, runtime, and hardware teams to deliver product improvements.
Summary Generated by Built In

SambaNova is a leader in next-generation AI infrastructure, delivering a full-stack inference platform for customers worldwide. At the core of SambaNova's technology is the RDU (Reconfigurable Dataflow Unit) — a chip built on a dataflow architecture rather than the traditional GPU model. Its decode performance is especially strong for agentic workloads like multi-turn agents, code generation, and long-running applications. RDUs are packaged into SambaRack, rack-scale hardware that lets customers deploy state-of-the-art models with better performance, greater energy efficiency, and faster time to value.

About the Team

The AI Performance team brings up and optimizes foundation models such as DeepSeek, Llama, Qwen, and GPT OSS on SambaNova's dataflow platform. We profile and tune models across the compiler, runtime, and hardware layers to improve inference throughput and latency.

About the Role

As a new grad AI Performance Engineer, you'll bring up new open source models on the SambaNova stack and work to make them run faster. You'll profile model execution, find bottlenecks, and apply optimizations such as quantization, kernel fusion, and improved scheduling with the compiler, runtime, and hardware teams.

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.

Skills Required

  • Bachelor’s degree in Computer Science, Electrical Engineering, or a 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
  • Master’s or Ph.D. in Computer Science, Electrical Engineering, or a 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
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Palo Alto, CA
500 Employees
Year Founded: 2017

What We Do

AI is changing the world and at SambaNova, we believe that you don’t need unlimited resources to take advantage of the most advanced, valuable AI capabilities - capabilities that are helping organizations explore the universe, find cures for cancer, and giving companies access to insights that provide a competitive edge. We deliver the world’s fastest and only complete AI solution for enterprises and governments with world-record inference performance and accuracy. Powered by the SambaNova SN40L Reconfigurable Dataflow Unit (RDU), organizations can build a technology backbone for the next decade of AI innovation with SambaNova Suite. Our fully integrated hardware-software system, DataScale®, enables organizations to train, fine-tune, and deploy the most demanding AI workloads using the largest and most challenging models. Most recently, with the launch of our newest offering, SambaNova Cloud, developers can supercharge AI-powered applications on Llama 3.2 models. SambaNova was founded in 2017 in Palo Alto, California, by a group of industry luminaries, business leaders, and world-class innovators who understand AI. Today, we’ve built an incredibly smart and motivated team dedicated to making a lasting impact on the industry and equipping our customers to thrive in the new era of AI.

Why Work With Us

As a talent first company, we aim to hire the greatest and most innovative minds in the industry- driving the next generation of AI computing where no barrier is too high and the possibilities are truly limitless. We encourage our peers to take risks and take the initiative to make a lasting impact on the AI and ML industries.

Gallery

Gallery

Similar Jobs

Liberty Mutual Insurance Logo Liberty Mutual Insurance

Senior Underwriting Manager, Liberty Mutual Mobility Solutions

Artificial Intelligence • Fintech • Insurance • Marketing Tech • Software • Analytics
Remote or Hybrid
5 Locations
40000 Employees
98K-359K Annually

Liberty Mutual Insurance Logo Liberty Mutual Insurance

Senior Technical Claims Specialist, Commercial Excess and Coverage

Artificial Intelligence • Fintech • Insurance • Marketing Tech • Software • Analytics
Remote or Hybrid
5 Locations
40000 Employees
83K-174K Annually

Optum Logo Optum

Senior Business Analyst

Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
Hybrid
San Antonio, TX, USA
160000 Employees
73K-130K Annually

Optum Logo Optum

Community Pharmacy Manager

Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
In-Office
Waxahachie, TX, USA
160000 Employees
113K-193K Annually

Similar Companies Hiring

Unusual Machines, Inc. Thumbnail
Hardware • Robotics
Orlando, FL
300 Employees
Blee Thumbnail
Artificial Intelligence • Marketing Tech • Software
New York, New York
33 Employees
Vega Thumbnail
Artificial Intelligence • Automotive • Insurance • Transportation
US
65 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account