We are looking for experienced engineers to help build and scale next-generation AI infrastructure using Py Torch, one of the world’s most widely used deep learning frameworks. This role sits at the intersection of machine learning systems, compilers, and high-performance computing, enabling researchers and product teams to train and deploy large-scale models efficiently. You will work on core components of the Py Torch ecosystem, including model execution, distributed training, performance optimization, and developer experience.

What you'll be doing:

• Design and build core Py Torch capabilities across runtime, autograd, distributed training, and model execution

• Optimize performance across GPU/accelerator backends (CUDA, Triton, etc.)

• Contribute to or lead development of large-scale ML systems and infrastructure

• Improve model training efficiency, scalability, and reliability across multi-node environments

• Work on compilers / graph transformations / kernel optimizations to accelerate deep learning workloads

• Partner with researchers and applied teams to translate cutting-edge models into production systems

• Drive open-source contributions and collaborate with the broader Py Torch community

• Influence roadmap and architecture for next-gen AI platforms

• Work at the forefront of AI and accelerated computing

• Direct impact on how Py Torch runs on the world’s most advanced GPU platforms

• Collaborate across hardware, systems software, and AI research to push performance boundaries and enable breakthroughs in generative AI, autonomous systems, and high-performance computing

What we need to see:

• PhD or MSc degree in Computer Science, Applied Math, Physics, or related science or engineering field (or equivalent experience)

• 8+ years of software development experience,

• Strong programming skills in C++ and Python

• Deep understanding of deep learning frameworks, preferably Py Torch

• Experience with GPU programming (CUDA or similar) and performance optimization

Ways to stand out from the crowd:

• Contributions to Py Torch core or ecosystem libraries

• Experience with NVIDIA AI stack (TensorRT, Triton Inference Server, cuBLAS, cuDNN, NCCL)

• Familiarity with ML compilers (Torch Inductor, Triton, XLA, TVM)

• Experience optimizing LLMs or large-scale recommendation / vision models

• Background working closely with hardware-aware software optimization

With competitive salaries and a generous benefits package, we are widely considered to be one of the technology world’s most desirable employers. We have some of the most forward-thinking and hardworking people in the world working for us and, due to outstanding growth, our best-in-class engineering teams are rapidly growing.

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 184,000 USD - 287,500 USD for Level 4, and 224,000 USD - 356,500 USD for Level 5.

You will also be eligible for equity and benefits.

Applications for this job will be accepted at least until April 27, 2026.

This posting is for an existing vacancy.

NVIDIA uses AI tools in its recruiting processes.

NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.