Senior Manager, Software Engineering - JAX

NVIDIA
United States
Workplace: OnsiteFull timeUSD 272,000 - 431,250 annuallyFunction: Software EngineeringExperience: 10+ yearsEducation: mastersSkills: ["Leadership","Mentorship","Cross-functional collaboration","Strategic planning","Communication"]

Lead NVIDIA’s JAX engineering strategy by coordinating multiple teams to deliver peak performance across heterogeneous systems spanning GPUs, CPUs, and LPUs. Drive contributions across the JAX ecosystem (core, XLA, Shardy, MPMD, and application stacks like MaxText), build cross-org processes and success criteria, and make strategic staffing and hiring decisions while mentoring technical leaders. Partner with key open-source stakeholders to influence project direction and accelerate training, inference, and robotics readiness on next-gen architectures.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
NVIDIA
NVIDIA
4 months ago

Senior Manager, Software Engineering - JAX

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 29 minutes agoStatus: Live

Job Summary

Lead NVIDIA’s JAX engineering strategy by coordinating multiple teams to deliver peak performance across heterogeneous systems spanning GPUs, CPUs, and LPUs. Drive contributions across the JAX ecosystem (core, XLA, Shardy, MPMD, and application stacks like MaxText), build cross-org processes and success criteria, and make strategic staffing and hiring decisions while mentoring technical leaders. Partner with key open-source stakeholders to influence project direction and accelerate training, inference, and robotics readiness on next-gen architectures.
Location: United States
Workplace: Onsite
Employment Type: Full time
Job Function: Software Engineering
Seniority: Manager level

Key Responsibilities

  • •Drive engineering contribution strategy across the JAX ecosystem stack, prioritizing impact on performance, adoption, and ecosystem influence.
  • •Promote teamwork across organizational boundaries across compiler, runtime, networking, and hardware teams.
  • •Build deep partnerships with key open-source projects, especially Google’s JAX team, representing NVIDIA’s interests.
  • •Design processes and define clear success criteria to keep teams aligned to outcomes.
  • •Lead, grow, and mentor a high-performing engineering organization, including staffing plans, capacity review, and strategic hiring decisions.

Pay and Benefits

Salary: USD 272,000 - 431,250 annually
Equity and Bonus:Equity

Key Requirements

  • •MS or PhD in Computer Science, Computer Engineering, or a related field (or equivalent experience).
  • •10+ 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 spanning organizational boundaries, influencing without direct authority.
  • •Track record of owning outcomes across multiple projects and teams, balancing trade-offs and driving progress with incomplete information.
Experience:10+ yearsAI researchHigh-performance computingCompilersDeep learning frameworksOpen sourceHeterogeneous compute
Education:Master's
Skills:LeadershipMentorshipCross-functional collaborationStrategic planningCommunication
Tech Stack:JAXXLAShardyMPMDMaxTextTileIRCUDAPJRTHLOPyTorchTensorFlowGPU programmingCuDNNNCCLTensorRTTransformer EngineLPUsRubin UltraRLHFHeterogeneous compute

Company Brief

NVIDIA
Designs and manufactures GPUs, AI accelerators, and system-on-chip products for gaming, data centers, professional visualization, and automotive markets, enabling advanced graphics, AI, and high-performance computing solutions worldwide.
Industry: Electronics Manufacturing
Company Size: Enterprise (1,001+ employees)
Revenue: USD 1B+
Growth: Public Company
Valuation: Public Company (Market Cap in USD)
Funding: IPO / Publicly Listed
Headquarters: Santa Clara, United States
Founded: 1993
Glassdoor
Glassdoor: 4.3
WebsiteLinkedInGlassdoor