Senior AI Performance Engineer
SambaNova SystemsBase Salary
$180k - $255k/yr
Responsibilities
- Bring up and optimize foundation models including DeepSeek, Llama, and Qwen on the SambaNova platform.
- Profile and enhance model performance across compiler, runtime, and hardware layers to improve throughput and latency.
- Collaborate with machine learning, compiler, runtime, and hardware teams on high-performance AI applications.
- Apply advances in model architecture, quantization, scheduling, and memory optimization.
- Develop scalable, efficient end-to-end inference solutions aligned with customer needs.
- Identify bottlenecks and develop dataflow or scheduling optimizations for single-node and distributed systems.
Requirements
- Bachelor’s or higher degree in computer science, electrical engineering, applied mathematics, physics, statistics, or a related field.
- At least 3 years of experience in deep learning model development and performance optimization, compiler/runtime/kernel optimization, software-hardware co-design, or systems performance tuning.
- Proficiency in Python or C++ with strong foundations in algorithms, data structures, and numerical computing.
- Experience with at least one major ML framework, including PyTorch, TensorFlow, or JAX.
- Demonstrated ability to analyze and optimize real-world ML pipelines.
- Preferred qualifications include LLM or multimodal model training and inference, distributed training, continuous batching, high-throughput inference, quantization, graph optimization, kernel fusion, and model partitioning.
- Preferred experience includes DeepSpeed, Megatron, vLLM, TensorRT, CUDA, Triton, OpenCL, cuDNN, cuBLAS, memory hierarchy optimization, caching, scheduling, publications, or open-source contributions.
Benefits
- Base salary range of $180,000—$255,000 USD, plus equity and benefits.
- US-based full-time employees receive medical insurance with 95% employee premium coverage and 77% dependent premium coverage, plus an employer-contributed HSA.
- Benefits include dental, vision, short- and long-term disability, basic and voluntary life insurance, AD&D, and flexible spending accounts.
- Well-being benefits include Headspace, Gympass+ with physical gym access, One Medical, counseling services, and an Employee Assistance Program.
Categories
About SambaNova Systems
Welcome to SambaNova: Revolutionizing AI Capacity At SambaNova, we're empowering developers, enterprises, governments, and data centers to unlock their full AI potential. Our full-stack infrastructure, from chips to models, enables lightning-fast performance, low power consumption, and high-efficiency computing. Our Mission To give every developer, enterprise, government and data center absolute sovereignty over their own data, models and AI infrastructure – to future-proof the AI workloads that will power and scale tomorrow. Our Technology We give our customers the optionality to experience SambaNova through the cloud or on-premise. Samba Cloud delivers the fastest inferences on the largest open source models like Llama 4 and DeepSeek. Developers can get started building in minutes with our OpenAI compatible APIs. All customers start on the developer tier and when they need more capacity can scale into our enterprise tier. SambaStack is our on-premise offering which includes the system, the platform, and foundation models. These components combine into a powerful technology stack that delivers unparalleled performance, ease of use, accuracy, data privacy, and the ability to power every use case across the world's largest organizations. SambaManaged is a modular and ready-to-deploy AI cloud designed to deliver unmatched efficiency for data centers and cloud service providers. This solution allows organizations to quickly deploy advanced AI inference services—without the need for costly infrastructure upgrades or specialized expertise—in as little as 90 days. At the heart of SambaNova innovation is the Reconfigurable Dataflow Unit (RDU). Purpose built for AI workloads, the RDU takes advantage of a dataflow architecture and a three-tiered memory design. The three tiers of memory enable the platform to run hundreds of models on a single node and to switch between them in microseconds. In 2023, SambaNova released its 4th generation RDU chip, the SN40L.