Senior Deep Learning Performance Engineer

We are now looking for a Senior Deep Learning Performance Engineer!

NVIDIA is seeking highly skilled and dedicated engineers who are passionate about optimizing AI workloads, particularly Generative AI and Large Language Model (LLM) training. This role requires working across all levels of the hardware and software stack, including GPU architecture, compilers, kernels, and Deep Learning frameworks such as JAX and PyTorch. You'll have the opportunity to work on cutting-edge technology that will accelerate training performance for deep learning users all over the world. Are you ready to take on challenging problems and collaborate with innovative teams to engineer systems for high-performance Deep Learning? This could be the role for you!

What You Will Be Doing:

• Understand, analyze, profile, and optimize large language model training on state-of-the-art hardware and software platforms.

• Understand the big picture of L...