Senior AI Platform Engineer - JAX & Performance

NVIDIA is hiring senior engineers to develop its AI platform and optimize deep learning frameworks using JAX. You will join a fast, modular, and scalable effort focused on data, training, and analysis for diverse DL solutions.

You will design core JAX components, drive peak performance on NVIDIA hardware, collaborate with AI researchers, and build tools to increase team efficiency while bridging numerical computing and real-world applications.