All vacancies
Senior Manager, Software Engineering - JAX
NVIDIA
US, CA, Santa Clara · On-site · relocation$22,667–35,938 gross/mofull-timeAI model trainingNVIDIA Careers (offices abroad)
A critical part of this role is ensuring JAX is ready on next-generation architectures, including Rubin Ultra, LPUs, and the CPUs that drive them, from day one. Doing that well requires keeping up with where AI is heading.
Responsibilities
- Drive the engineering contribution strategy across the JAX ecosystem stack — including JAX core, XLA, Shardy, MPMD, and application-level codebases like MaxText — prioritizing where NVIDIA's contributions create maximum impact on performance, adoption, and ecosystem influence
- Build deep partnerships with key open-source projects, especially Google's JAX team — supporting our partners while representing NVIDIA's interests in shaping the project's direction
- Design processes and define clear success criteria to keep teams aligned to outcomes
- Build staffing plans, review team capacity, and make strategic hiring decisions to stay ahead of current and future needs
- Lead, grow, and mentor a high-performing engineering organization — developing technical leaders and creating space for your team to experiment, prototype, and build innovative systems that improve products and processes
Languages
- Work format
- On-site
- Seniority
- Lead
- Posted
- 3 Aug 2026
- Last verified
- In the last 5 days
Similar roles
- ChatGPT DeveloperKDCI · Remote · worldwide, UTC+4 accepted
- Senior Full Stack Developer (Node.js + PHP)Ready · Remote · worldwide, UTC+4 accepted
- Data DevOps LeadEPAM Systems · Remote · Armenia
- Banco de Talentos: Data Platform EngineerArco Educação · Remote · worldwide, UTC+4 accepted
- Senior SAP ABAP / UI5 DeveloperPontoonglobal · Remote · worldwide, UTC+4 accepted
- ASIC Digital Design, Sr EngineerSynopsys · Armenia · On-site