Senior Manager, Software Engineering - JAX at NVIDIA

You are a seasoned engineering manager with an MS or PhD and over a decade of systems software experience. You possess deep expertise in compilers, high-performance computing, or AI frameworks and have at least four years of experience leading complex technical teams. **What makes it worth a look...** NVIDIA is hiring for this full-time role in Santa Clara, offering an annual base salary range of $272,000 to $431,250 plus equity and comprehensive benefits. It is a rare chance to lead the strategy for JAX and influence next-generation hardware deployments across the largest supercomputers globally. **You might be a good fit if you...** * Have direct experience with JAX, XLA, or similar deep learning compiler infrastructures. * Understand GPU programming, CUDA, and heterogeneous compute targets. * Led large-scale projects involving hardware and software integration across cross-functional teams. * Possess a technical background in distributed AI training or inference performance optimization.