NVIDIA
Senior Manager, Software Engineering - JAX
$272K–$431K
US, CA, Santa Clara · Manager · Full-time · Posted Aug 3, 2026
- Experience
- 10+ overall years in software engineering, with 4+ years in engineering management or technical leadership.
- Education
- MS or PhD in Computer Science, Computer Engineering, or a related field (or equivalent experience).
About the job
Leads NVIDIA’s JAX engineering strategy and teams to deliver performance across heterogeneous platforms and next-generation architectures, while coordinating with internal teams and open-source partners.
What you'll do
- Set contribution priorities across the JAX ecosystem and coordinate work across compiler, runtime, networking, and hardware teams.
- Build open-source partnerships, define team processes and success criteria, and make staffing and hiring plans.
- Lead, mentor, and grow engineering teams delivering JAX performance and architecture support.
What they ask for
- MS or PhD in Computer Science, Computer Engineering, or a related field (or equivalent experience).
- 10+ years of software engineering experience, including 4+ years in engineering management or technical leadership.
- Strong technical background in system software, compilers, high-performance computing, or AI frameworks.
- Experience leading cross-organizational strategy and projects, owning outcomes across teams, improving workflows, and leading work with external collaborators.
Nice to have
- Hands-on experience with JAX and systems such as XLA, HLO, PJRT, or Shardy.
- Experience with compiler/runtime infrastructure, NVIDIA AI platform components, large-scale AI workload optimization, or bringing software up on new hardware.
Pay
272,000 USD - 431,250 USD base salary
Tools
JAX, XLA, Shardy, MPMD, MaxText, TileIR, CUDA, PJRT
AI focus
Enables JAX for emerging AI needs in training, post-training, inference, and robotics, including performance across GPUs, CPUs, and LPUs.
Model-written summary · the original posting is authoritative