Senior JAX Engineer: AI Platform & Performance
Listed on 2026-11-01
-
Software Development
AI Reliability/ Performance Engineer, Senior Developer
NVIDIA is seeking senior engineers to advance its AI platform, focusing on performance optimizations in deep learning frameworks using JAX. You will design core JAX components, optimize CPU/GPU performance, and collaborate with AI researchers to deliver scalable, modular tooling for data, training, and analysis across diverse DL workloads.
The role requires strong programming skills, system design expertise, and excellent communication to coordinate across teams and geographies.
As a Senior JAX Engineer: AI Platform & Performance, you will play an important part at Nvidia in United States.
The advertised compensation is 150..
We aim to respond to suitable candidates as soon as possible.
Full responsibilities and requirements are described in the listing above.
Learn more about the Senior JAX Engineer: AI Platform & Performance role in the description above.
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).