Senior Manager, Software Engineering - Jax

Nvidia Corporation

CA, United States
Base: 272,000 usd - 431,250 usd; bonus/equity: equ...
**
Drive jax strategy across ecosystem
Promote teamwork across boundaries
Build partnerships with open-source projects
** Nvidia is seeking a Senior Manager of Software Engineering for their JAX framework, focusing on AI research and performance optimization across heterogeneous computing architectures. The ideal candidate will have extensive software engineering experience and a strong background in system software, compilers, or AI frameworks. **

Job Summary

  • Drive the engineering contribution strategy across the JAX ecosystem stack, prioritizing where NVIDIA's contributions create maximum impact.
  • Build deep partnerships with key open-source projects, especially Google's JAX team, while representing NVIDIA's interests.
  • Lead, grow, and mentor a high-performing engineering organization, developing technical leaders and creating space for innovation.

Matching Summary

Match Score: 75

** Nvidia is seeking a Senior Manager of Software Engineering for their JAX framework, focusing on AI research and performance optimization across heterogeneous computing architectures. The ideal candidate will have extensive software engineering experience and a strong background in system software, compilers, or AI frameworks. **

Salary

Base: 272,000 USD - 431,250 USD; Bonus/Equity: Equity; Benefits: Benefits

Skills & Requirements

Must-have

  • Drive JAX strategy across ecosystem
  • Promote teamwork across boundaries
  • Build partnerships with open-source projects
  • Design processes and success criteria
  • Build staffing plans and hiring decisions
  • Lead and mentor engineering organization

Nice-to-have

  • Experience with JAX systems
  • Compiler or runtime infrastructure experience
  • Familiarity with NVIDIA AI platform
  • Drive performance optimization efforts
  • Drive framework bring-up on new hardware

Key Requirements

  • MS or PhD in Computer Science or related field
  • 10+ years software engineering experience
  • 4+ years engineering management or technical leadership
  • Strong technical background in system software, compilers, HPC, or AI frameworks
  • Proven experience leading domain-level strategy
  • Track record of owning outcomes across multiple projects

Work Rights

Not specified

Tailored Resume

Cover Letter