Senior AI Platform Engineer - JAX & Tools
Listed on 2026-10-30
-
Software Development
AI Engineer (Applied/Software), AI Reliability/ Performance Engineer, Senior Developer
NVIDIA seeks senior engineers to optimize its AI platform using JAX. You will design and implement JAX core components, driving peak performance on NVIDIA hardware and collaborating with AI researchers to build scalable models.
The role requires deep expertise in C/C++ and Python, strong system design and communication skills, and the ability to ship customer-facing solutions across multi-functional teams. This posting reflects a high-expectation, fast-moving environment.
This role is for the Senior AI Platform Engineer - JAX Performance & Tools role at NVIDIA.
We are looking to fill the Senior AI Platform Engineer - JAX Performance & Tools position at NVIDIA in CA, United States.
The Senior AI Platform Engineer - JAX Performance & Tools role at NVIDIA is now open for applications in CA, United States.
Join us at NVIDIA as our next Senior AI Platform Engineer - JAX Performance & Tools in CA, United States.
(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).