reqspace
Start free trial →
N
Nvidia

Senior Manager, Software Engineering - JAX

USonsitemanager

Posted today · via Workday

See if I'm a fit →Tailor my resume for this role →Apply on Workday ↗

About this role

JAX is the framework of choice for the most ambitious AI research, and the datacenters it runs on are evolving fast. NVIDIA supercomputers are becoming more heterogeneous combining GPUs, CPUs, and LPUs, and JAX needs to evolve to perform well across all of them. NVIDIA seeks a Senior Engineering Manager to define and drive NVIDIA's JAX strategy, coordinating multiple teams and workstreams to ensure JAX delivers peak performance across the platform, from single-accelerator workloads to hundreds of thousands of devices on the largest supercomputers ever built. A critical part of this role is ensuring JAX is ready on next-generation architectures, including Rubin Ultra, LPUs, and the CPUs that drive them, from day one. Doing that well requires keeping up with where AI is heading.…

Read the full description on Nvidia's site →

What we'd score you on

reqspace match rubric

Five dimensions, recruiter-grade. Upload your resume and we'll generate a written explanation of where you fit and where the gaps are.

1

Skills match

For this role: pytorch, tensorflow, jax, teams

2

Level fit

This role is manager-level. We check your trajectory against it.

3

Domain experience

Your work in the role's domain matters more than your years total. We weight recent and direct experience.

4

Recency

A skill you used last quarter weighs more than one from five years ago. We grade on recency, not lifetime.

5

Location fit

This role is based in US. We weight your proximity and willingness to relocate.

Score yourself on this role.
Free · no card · written explanation included
See if I'm a fit →

Skills in this role

Pulled from the job description. These are the keywords we'll weight when scoring your fit.

pytorchtensorflowjaxteams

More at Nvidia

See all open jobs at Nvidia →