Senior Manager, Software Engineering - JAX - CA, Santa Clara, United States
Originally posted 3 August 2026 by the employer. Relisted by the employer on 7 August 2026.
Lead NVIDIA's engineering strategy across the JAX stack, targeting XLA, Shardy, MPMD, and heterogeneous supercomputers.
About the role
NVIDIA seeks a Senior Engineering Manager to define and drive NVIDIA's JAX strategy across heterogeneous platforms spanning GPUs, CPUs, and LPUs. The role coordinates teams across the JAX ecosystem stack, including JAX core, XLA, Shardy, MPMD, and Rubin Ultra architectures.
What you'll do
- Drive the engineering contribution strategy across the JAX ecosystem stack, including JAX core, XLA, Shardy, MPMD, and application-level codebases like MaxText. - Work across organizational boundaries to align compiler (XLA, TileIR, CUDA), runtime, networking, and hardware teams. - Build partnerships with open-source projects, including Google's JAX team, to represent NVIDIA's direction. - Design processes and define clear success criteria to keep engineering teams aligned to technical outcomes. - Build staffing plans, review team capacity, and make strategic hiring decisions. - Lead, grow, and mentor an engineering organization, developing technical leaders to build and prototype systems.
What you'll need
- 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. - Technical background in system software, compilers, high-performance computing, or AI frameworks. - Experience leading domain-level strategy and projects that span organizational boundaries, building consensus, and communicating from engineers to executives. - Track record of owning outcomes across multiple projects and teams while balancing trade-offs. - Experience identifying development cycle gaps and defining workflows across teams and organizations. - Experience leading teams that contribute to large-scale projects with external collaborators.
Skills: JAX, XLA, MPMD, supercomputers, AI research, open-source projects
This role has been open 202 days — well above the 64-day median for Software Engineering Semiconductor roles.
Software Engineering Semiconductor · AI ML Hardware
|
Open roles in category
152
|
Median days open
64 d
|
Median salary
$188k
|
See the full market breakdown ▾Category comparison, skills in demand, and who else is hiring
| Metric | NVIDIA | All employers we track in this specialty (152 roles · 42 employers) |
|---|---|---|
| Open roles in this specialty | 26 | 152 |
| Open roles in the wider Software, Firmware & Systems family | 1012 | 6328 · 146 employers |
| Median days open | 214 d | 64 d (+150 d vs this employer) |
| Median salary (USD postings) | — | $188k |
Skills observed across this category: JAX, XLA, MPMD, supercomputers, AI research, open-source projects
Who's hiring in this category
- Micron Technology · 25 open roles · median 42 d
- NVIDIA (this employer) · 25 open roles · median 226 d
- Qualcomm · 20 open roles · median 202 d
- Infineon Technologies · 12 open roles · median 47 d
- AMD · 8 open roles · median 64 d
- Applied Materials · 4 open roles · median 98 d
How we counted: 152 open Software Engineering Semiconductor (AI ML Hardware) roles from 42 employers tracked in the SemiconductorJobs index, counted 28 Oct 2026. Specialty figures count only roles carrying this exact specialty label, so an employer's related work in neighbouring specialties is not included there — it is counted in the wider Software, Firmware & Systems family row. Figures refresh nightly.