AI Performance Engineer
Applied Intuition - Sunnyvale, CA
Hiring: AI Performance Engineer Company: Applied Intuition Location: Sunnyvale, CA Job Posted Time: 2026-09-10 10:28:59 Employment Type: Full-time / Remote Target Skills & Keywords : C++, Kubernetes, Machine Learning, Node.js, ONNX, OpenCV, PyTorch, Python, TensorRT, Triton About the job Required Skills: •Profile and optimize distributed training end to end - data loading and preprocessing, augmentation, kernel execution, gradient communication, and checkpointing •Optimize large-scale offline and batch inference over petabyte-scale sensor logs: batching and scheduling strategies, quantization and low-precision execution, graph optimization, and accelerator saturation across long-running sweeps •Establish roofline and performance models for our workloads, quantify the gap between achieved and theoretical performance, and stack-rank optimization opportunities by impact and effort •Improve multi-node scaling efficiency: sharding and parallelism strategies, collective communication, interconnect utilization, and memory-bandwidth and kernel-fusion bottlenecks •Drive cluster goodput - reduce GPU idle time from input pipeline stalls, storage and network I/O, scheduling gaps, stragglers, and failure recovery on long-running jobs •Build the benchmarking, observability, and regression-detection tooling that keeps performance from silently degrading as models and code evolve •Partner cross-functionally with engineers across functions to solve complex data and compute problems at scale •Contribute to a team culture that values effective collaboration, technical excellence, and innovation Qualifications: •GPU kernel development experience: CUDA, Triton, CUTLASS, or hand-tuned attention implementations •Operational familiarity with autonomy or robotics data (ROS, OpenCV, multi-sensor log formats) •Don’t meet every single requirement? If you’re excited about this role but your past experience doesn’t align perfectly with every qualification in the job description, we encourage you to apply anyway. You may be just the right candidate for this or other roles. Compensation: •$215,000 - $285,000 / year •Flexible work environment (work from home / hybrid options) Interested candidates, please apply directly through the job posting on company's career page or try via AI auto apply on this platform. Don't miss this opportunity to join a forward-thinking team!