Senior Manager, Software Engineering - JAX
Core
Define and drive NVIDIA's JAX strategy to ensure peak performance across heterogeneous supercomputing platforms from single-accelerator to hundreds of thousands of devices.
Role type
Senior Manager, Software Engineering
Builds
NVIDIA's JAX ecosystem stack (JAX core, XLA, Shardy, MPMD, MaxText) and AI platform across compiler, runtime, and networking
Domain
AI Research / High-Performance Computing / Supercomputing
Deliverable
production ML models
Required skills
Engineering management, technical strategy, cross-organizational leadership, open-source partnership, hiring, team building
Preferred skills
JAX internals (XLA, HLO, PJRT), compiler/runtime infrastructure, heterogeneous compute (GPU/CPU/LPU), distributed AI training/inference
Technologies
JAX, XLA, Shardy, CUDA, NCCL, TensorRT, Transformer Engine, Rubin Ultra, LPUs
Responsibilities
Drive engineering contribution strategy across the JAX ecosystem stack; Promote teamwork across organizational boundaries; Build deep partnerships with key open-source projects; Design processes and define success criteria; Build staffing plans and make strategic hiring decisions; Lead, grow, and mentor a high-performing engineering organization
Seniority
Senior, hands-on IC manager
