Senior Staff Machine Learning Performance Engineer

What you'll do

• Optimize training step time, scaling efficiency, and stability across data pipelines, kernels, parallelism, and communication.
• Lead performance work with modeling, infrastructure, simulation, data-mining, and robotics teams; turn improvements into reusable defaults.
• Expand low-overhead performance observability and build agents to reproduce jobs, verify numerical parity, and detect regressions.

What they ask for

• Master's or PhD in CS, CE or EE, or equivalent practical experience.
• Hands-on experience optimizing GPU training or inference workloads in production at multi-node scale.
• Demonstrated ability to diagnose and fix performance issues in production, measuring and verifying gains.
• In-depth computer architecture and high-performance computing knowledge; strong C++ and Python and solid data structures and algorithms skills.

Nice to have

• Experience writing and tuning CUDA or Triton kernels; fluent in PyTorch.
• Experience with profiling, tracing, or benchmarking infrastructure, LLM agents, large-scale training, or autonomous driving/robotics.

Pay

USD $244,140, - $413,160 base salary

Tools

C++, Python, CUDA, Triton, PyTorch, CUPTI, Perfetto

AI focus

Optimize training for autonomous-driving foundation models, including vision-language-action policies and world models; build agents that automate performance optimization and regression detection.