- MS or PhD in Computer Science, Computer Engineering, or a related field (or equivalent experience)
- 10+ overall years of software engineering experience, with 4+ years in engineering management or technical leadership
- Strong technical background in system software, compilers, high-performance computing, or AI frameworks
- Proven experience leading domain-level strategy and projects that span organizational boundaries, building consensus, influencing without direct authority, and tailoring communication to the audience from engineers to executives
- Track record of owning outcomes across multiple projects and teams, balancing trade-offs, and driving progress with incomplete information
- Experience identifying development cycle gaps and defining workflows across teams and organizations to improve efficiency
- Experience leading teams that contribute to large-scale projects with external collaborators
Ways to stand out from the crowd
- Hands-on experience with JAX and its systems like XLA, HLO, PJRT, Shardy, or JAX's tracing and transformation system
- Experience leading teams working on compiler or runtime infrastructure for deep learning frameworks (JAX, PyTorch, TensorFlow or equivalent)
- Familiarity with NVIDIA's AI platform components: GPU programming, CUDA, cuDNN, NCCL, TensorRT, Transformer Engine and heterogeneous compute targets (GPU, CPU, LPU)
- Experience driving performance optimization efforts for large-scale distributed AI training, post-training (RLHF, alignment), inference, or robotics workloads (multi-node, multi-GPU)
- Experience driving framework or software bring-up on new hardware architectures, coordinating across silicon, driver, compiler, and framework teams
Your base salary will be determined based on your location, experience, and other objective job-related factors. The expected base salary range for this position is 272,000 USD – 431,250 USD.
You will also be eligible for equity and benefits.
Applications for this job will be accepted at least until November 3, 2026.
This posting is for an existing vacancy.
NVIDIA uses AI tools in its recruiting processes.