Senior Manager, Software Engineering - JAX

Requirements:

• MS or PhD in Computer Science, Computer Engineering, or 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
• Experience identifying development cycle gaps and defining workflows across teams
• Experience leading teams that contribute to large-scale projects with external collaborators

What you'll be doing:

• Drive the 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 clear success criteria
• Build staffing plans, review team capacity, and make strategic hiring decisions
• Lead, grow, and mentor a high-performing engineering organization

Nice to haves:

• Hands-on experience with JAX and its systems
• Experience leading teams working on compiler or runtime infrastructure for deep learning frameworks
• Familiarity with NVIDIA's AI platform components
• Experience driving performance optimization efforts for large-scale distributed AI training
• Experience driving framework or software bring-up on new hardware architectures