Principal Product Manager, AI Frameworks

Full-time other-general

At NVIDIA, we meet customers where they are on their AI journey on our GPUs - this means we build best in class frameworks in OSS and support a robust ecosystem of other OSS frameworks. NVIDIA's PyTorch Compilers team builds and upstreams to the stack that sits between PyTorch and NVIDIA hardware — spanning torch.compile, emerging compiler substrates, and the agent-native optimization infrastructure being built for the next era of accelerated computing. This role will build and direct product strategy across the full arc. It involves shipping the latest hardware features in torch.compile today. It also includes developing the canonical shared representation across NVIDIA's compiler and runtime ecosystem. Additionally, it focuses on crafting how agents will engage in deep learning performance work in the future.


We are looking for someone who understands compilers and can operate at the intersection of systems architecture, framework engineering, and customer-facing pr...