Senior Technical Program Manager, Deep Learning Frameworks

NVIDIA's Deep Learning Software team is seeking a Technical Program Manager to lead software programs for Deep Learning Training and Inference Frameworks. The goal is to support NVIDIA's top AI researchers and software engineers in driving the future of deep learning.

The TPM will manage software programs in a highly matrixed environment, navigating an active and evolving roadmap. They will work with senior management, partners, and teams across the company to ensure alignment and exceptional execution.

Responsibilities:

• Collaborate with hardware, software, and model program managers, product managers, and engineering teams to deliver Deep Learning Frameworks programs on existing and new hardware.
• Engage with cross-company teams and drive alignment on release scope, milestones, risk management, and dependencies.
• Guide software programs in all aspects of program management – planning, forecasting, documenting, scheduling, effective meetings, multi-faceted prioritization, management of dependencies, reporting, and effective handling of critical and blocking issues.
• Develop and implement metrics for measuring program effectiveness and improvement areas, collect and analyze data in support of planning and data-driven decisions.
• Define and implement standard processes for open-source contribution and release management within the Nvidia AI/ML ecosystem.
• Report on overall program status, providing insights and recommendations to senior management.
• Drive organizational alignment and efficiency by coordinating with multi-functional leads and streamlining processes.

Requirements:

• Postgraduate degree in Computer Science, Artificial Intelligence, Mathematics, or equivalent experience.
• 10+ years of software program management experience, including a proven track record of leading global projects in fast-paced software development environments.
• Proven experience with delivering large software programs spanning multiple layers of the software stack.
• Ability to think strategically and tactically, building consensus to make programs successful by engaging and moderating successful engagements with engineering and product teams.
• Excellent communication, technical presentation, and attention to detail skills, with a shown ability to multitask in a dynamic environment with shifting priorities and changing requirements.
• Excellent organizational skills and ability to use project management tools (e.g., Jira, Aha!, Confluence) and distributed version control systems (e.g., Git).

Preferred Qualifications:

• Experience with Deep Learning Frameworks (PyTorch, Jax, etc.), ML compilers (XLA, Triton, etc.), GPU Technology, open-source development.
• Prior experience in production software development, release management, DevOps, with a consistent track record of driving process improvements and measuring efficiency.
• Exposure to NVIDIA GPU programming and software stack (CUDA Toolkit, cuDNN, TensorRT, NCCL, etc.).
• Engineering background.

You will also be eligible for equity and benefits.