Senior Technical Program Manager, Deep Learning Frameworks

Deep Learning Software is looking for a Technical Program Manager to lead software programs for Deep Learning Training and Inference Frameworks, with the goal of supporting NVIDIA's top AI researchers and software engineers in driving the future of deep learning. This position requires managing software programs in a highly matrixed environment using industry-leading processes to navigate an active and evolving roadmap.

The TPM will work alongside senior management and coordinate efforts with partners across the company to ensure alignment and exceptional execution. They will engage closely with our software, hardware, and infrastructure teams to foster strong engineering collaborations. Furthermore, they will develop and standardize planning, reporting, and execution practices and metrics to achieve ambitious objectives in a rapidly changing environment.

• 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 (hardware/software engineering, product, QA, compliance) 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.

• Postgraduate degree in Computer Science, Artificial Intelligence, Mathematics or equivalent experience.
• 10+ years software program management experience including proven track record leading global projects, adaptable to multiple time zones 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 and to build 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).

• 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, Dev Ops with a consistent track record of driving process improvements and measuring efficiency.
• Exposure to NVIDIA GPU programming and software stack (CUDA Toolkit, cuDNN, Tensor
  
  RT, NCCL, etc.).
• Engineering background.

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 168,000 USD - 258,750 USD for Level 4, and 200,000 USD - 322,000 USD for Level 5. You will also be eligible for equity and benefits.

Applications for this job will be accepted at least until April 12, 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.