layiq
/la·yiq/adjective
worthy; deserving; fitting; suitable.
A role, opportunity, or path that merits attention, time, and pursuit.
Loading LAYIQ…
Job opportunity
Senior Manager, Software Engineering - JAX
NVIDIA
US, CA, Santa Clara
Source: NVIDIA careers · View original posting
- Pay
- $272,000 — $431,250 USD per year
- Employment
- full time
Employer job description
From NVIDIA's posting. “We” and “our” refer to the employer.
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. This means supporting emerging needs across training, post-training, inference, and robotics, bridging new hardware capabilities with AI trends. Come join us to build the team and technology behind the next AI breakthroughs.