Senior Technical Program Manager, Deep Learning Libraries

NVIDIA • Santa Clara, CA

3d

About The Position

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.

Requirements

• 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).

Nice To Haves

• 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.

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 (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.

What This Job Offers

• Job Type: Full-time
• Career Level: Senior