NVIDIA

3 weeks 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: 21 hours agoStatus: Live

Job Summary

Define and drive NVIDIA’s JAX strategy by coordinating multiple teams and workstreams across the JAX ecosystem stack, including JAX core, XLA, Shardy, MPMD, and application-level codebases like MaxText. Build cross-org processes and success criteria, establish staffing and hiring plans, and lead, mentor, and grow a high-performing engineering organization. Partner with key open-source projects, especially Google’s JAX team, to shape direction and improve performance across next-generation hardware.

Location: Germany, Munich

Workplace: Remote

Employment Type: Full time

Job Function: Software Engineering

Seniority: Sr. Manager level

Key Responsibilities

•Drive the engineering contribution strategy across the JAX ecosystem stack to maximize impact on performance, adoption, and ecosystem influence.
•Promote teamwork across organizational boundaries to shape NVIDIA’s AI platform across compiler, runtime, networking, and hardware teams.
•Build deep partnerships with key open-source projects, especially Google’s JAX team, representing NVIDIA’s interests in project direction.
•Design processes, define success criteria, and align teams to outcomes.
•Lead, grow, and mentor an engineering organization, including staffing plans, capacity review, and strategic hiring decisions.

Key Requirements

•10+ years of software engineering experience, including 4+ years in engineering management or technical leadership.
•MS or PhD in Computer Science, Computer Engineering, or a related field (or equivalent experience).
•Strong technical background in system software, compilers, high-performance computing, or AI frameworks.
•Proven experience leading domain-level strategy and projects across organizational boundaries, building consensus, influencing without authority, and communicating to engineers and executives.
•Track record owning outcomes across multiple projects and teams, balancing trade-offs, and driving progress with incomplete information.

Experience:10+ yearsAI researchHigh-performance computingDeep learning frameworksOpen sourceDistributed systemsRobotics

Education:Master's in Computer Science or Computer Engineering

Skills:LeadershipMentoringStrategic thinkingCross-functional collaborationCommunication

Tech Stack:JAXXLAShardyMPMDMaxTextTileIRCUDARuntimeNetworkingHLOPJRTJAX tracingJAX transformationPyTorchTensorFlowGPU programmingCuDNNNCCLTensorRTTransformer Engine

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: 4.3

Website LinkedIn Glassdoor

Loading job details...

Preparing the role view and application actions.

NVIDIA

3 weeks 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: 21 hours agoStatus: Live

Job Summary

Location: Germany, Munich

Workplace: Remote

Employment Type: Full time

Job Function: Software Engineering

Seniority: Sr. Manager level

Key Responsibilities

•Drive the engineering contribution strategy across the JAX ecosystem stack to maximize impact on performance, adoption, and ecosystem influence.
•Promote teamwork across organizational boundaries to shape NVIDIA’s AI platform across compiler, runtime, networking, and hardware teams.
•Build deep partnerships with key open-source projects, especially Google’s JAX team, representing NVIDIA’s interests in project direction.
•Design processes, define success criteria, and align teams to outcomes.
•Lead, grow, and mentor an engineering organization, including staffing plans, capacity review, and strategic hiring decisions.

Key Requirements

•10+ years of software engineering experience, including 4+ years in engineering management or technical leadership.
•MS or PhD in Computer Science, Computer Engineering, or a related field (or equivalent experience).
•Strong technical background in system software, compilers, high-performance computing, or AI frameworks.
•Proven experience leading domain-level strategy and projects across organizational boundaries, building consensus, influencing without authority, and communicating to engineers and executives.
•Track record owning outcomes across multiple projects and teams, balancing trade-offs, and driving progress with incomplete information.

Experience:10+ yearsAI researchHigh-performance computingDeep learning frameworksOpen sourceDistributed systemsRobotics

Education:Master's in Computer Science or Computer Engineering

Skills:LeadershipMentoringStrategic thinkingCross-functional collaborationCommunication

Tech Stack:JAXXLAShardyMPMDMaxTextTileIRCUDARuntimeNetworkingHLOPJRTJAX tracingJAX transformationPyTorchTensorFlowGPU programmingCuDNNNCCLTensorRTTransformer Engine

Company Brief

NVIDIA

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: 4.3

Website LinkedIn Glassdoor