Job
- Level
- Senior
- Job Field
- IT, Software, DevOps
- Employment Type
- Full Time
- Contract Type
- Permanent employment
- Location
- Munich
- Working Model
- Onsite
Job Summary
In this role, you will define the JAX strategy, coordinate teams, set performance goals, and support the integration of new architectures into the JAX ecosystem while leading and developing a high-performing engineering team.
Job Technologies
Your role in the team
- JAX is the framework of choice for the most ambitious AI research, and the datacenters it runs on are evolving fast.
- NVIDIA supercomputers are becoming more heterogeneous, combining GPUs, CPUs, and LPUs, and JAX needs to evolve to perform well across all of them.
- NVIDIA is seeking a Senior Engineering Manager to define and drive NVIDIA's JAX strategy, coordinating multiple teams and workstreams to ensure JAX delivers peak performance across the platform, from single-accelerator workloads to hundreds of thousands of devices on the largest supercomputers ever built.
- 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.
- This means supporting emerging needs across training, post-training, inference, and robotics, bridging new hardware capabilities with AI trends.
- 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.
- Promote teamwork across organizational boundaries to shape NVIDIA's AI platform across compiler (XLA, TileIR, CUDA), runtime, networking, and hardware teams.
- 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.
This text has been machine translated. Show original
Our expectations of you
Qualifications
- Strong technical background in system software, compilers, high-performance computing, or AI frameworks.
- Track record of owning outcomes across multiple projects and teams, balancing trade-offs, and driving progress with incomplete information.
Experience
- 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.
- 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.
- Experience in identifying gaps in the development cycle and defining workflows across teams and organizations to improve efficiency.
- Experience leading teams that contribute to large-scale projects with external collaborators.
This text has been machine translated. Show original
Topics that you deal with on the job
Job Locations
This is your employer
Nvidia
For the past two decades, NVIDIA has been constantly reinventing itself. The invention of the GPU by NVIDIA in 1999 sparked the growth of the PC gaming market, redefined modern computer graphics, and revolutionized parallel computing.
Description
- Founding year
- 1999
- Language
- English
- Company Type
- Established Company
- Working Model
- Full Remote, Hybrid, Onsite
- Industry
- Trade
Senior Manager, Software Engineering
- Location
- Munich
- Working Model
- Onsite
- Diversity
- Open for all genders
- English Only
- English only required
