NVIDIA is seeking a Senior Engineering Manager to lead the JAX strategy, focusing on optimizing performance across various architectures including GPUs, CPUs, and LPUs. This role involves coordinating multiple teams to ensure JAX is prepared for next-generation AI workloads. The manager will promote collaboration across organizational boundaries and build partnerships with key open-source projects. The position requires a strong technical background and leadership skills to drive innovation in AI frameworks.

• 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
• Strong technical background in system software, compilers, high-performance computing, or AI frameworks
• Proven experience leading domain-level strategy and projects that span organizational boundaries
• Track record of owning outcomes across multiple projects and teams, balancing trade-offs, and driving progress with incomplete information
• Experience identifying development cycle gaps and defining workflows across teams and organizations to improve efficiency
• Experience leading teams that contribute to large-scale projects with external collaborators

• Drive the engineering contribution strategy across the JAX ecosystem stack
• Promote teamwork across organizational boundaries to shape NVIDIA's AI platform
• Build deep partnerships with key open-source projects, especially Google's JAX team
• Design processes and define clear success criteria to keep teams aligned to outcomes
• Build staffing plans, review team capacity, and make strategic hiring decisions
• Lead, grow, and mentor a high-performing engineering organization

• Employees at NVIDIA are often offered comprehensive, day-one benefits—including medical, dental, and vision coverage with HSA support, life and disability insurance, an Employee Assistance Program, and a 401(k) with auto-enrollment. Many roles also have generous time off and holidays, donation matching (up to $10,000), and a wide menu of extras like FSAs, commuter benefits, legal and identity-theft protection, pet insurance, and wellness discounts. Optional programs can include student-loan and home-purchase support, plus family care resources and expert medical services.