Senior Deep Learning Compiler Engineer - PyTorch

Overview

Join to apply for the Senior Deep Learning Compiler Engineer - PyTorch role at NVIDIA

NVIDIA is seeking passionate engineers to build the next generation of tools used by AI developers and researchers worldwide. Our team is developing Thunder, an ambitious, source-to-source compiler built to unlock outstanding performance for PyTorch models on NVIDIA GPUs. This is a unique opportunity to contribute to a project that enhances the PyTorch ecosystem, working with modern compiler stacks like PyTorch 2.0's TorchDynamo and TorchInductor to create powerful, open-source solutions that benefit the entire community. If you are driven to solve complex problems and want to make a foundational impact on the AI ecosystem, apply to join our collaborative and innovative team.

What Youll Be Doing

As a key member of our team, you will be contributing directly to the future of accelerated AI. Your role will be dynamic and deeply technical, p...