Staff Software Engineer, Deep Learning Acceleration

What you'll do

• Analyze and optimize deep learning network performance on vehicles and in data centers.
• Improve software architecture, system performance, and latency; troubleshoot bottlenecks using profiling and the roofline model.
• Collaborate across teams to improve self-driving technology efficiency.

What they ask for

• At least 5 years of professional software engineering experience.
• BS, MS, or PhD in Computer Science or a related field.
• Strong programming skills in CUDA, C++, and Python; extensive high-performance computing and parallel-programming experience optimizing GPU memory usage, latency, or throughput.
• Experience with performance-analysis tools such as NVIDIA Nsight Systems and Nsight Compute, and framework-level optimization of deep-learning workloads using at least one framework.

Nice to have

• Hands-on experience in motion planning, robotics, autonomous systems, systems software, or computer vision.
• Experience with TensorRT, OpenAI Triton, Mojo, or other inference acceleration tools.

Pay

USD $189,000 - $274,000 per year (base salary), Mountain View, CA and San Francisco, CA; USD $171,000 - $247,000 per year (base salary), Pittsburgh, PA

Tools

CUDA, C++, Python, NVIDIA Nsight Systems, Nsight Compute, PyTorch, TensorFlow, TensorRT

AI focus

Accelerates deep-learning workloads for autonomous vehicles and data-center training, including model performance analysis, optimization, and deployment.