Outscal Logooutscal logo

Director of Product - AI Training Platform Software

1 Week ago • 10-8 Years • Artificial Intelligence • $272,000 PA - $425,500 PA

Job Summary

Job Description

As Director of Product - AI Training Platform Software at NVIDIA, you'll define the product strategy and roadmap for AI training software, encompassing LLMs and related frameworks. Collaborating with leading AI researchers and developers, you'll shape the future of AI training. This role involves partnering with external framework developers (PyTorch, JAX), driving go-to-market plans, and leading a team of product managers. You will need strong technical and product skills, deep expertise in AI training frameworks and LLMs, and excellent communication abilities. The position requires working from NVIDIA's Santa Clara headquarters and occasional travel.
Must have:
  • Develop product strategy and roadmap
  • Partner with AI researchers and developers
  • Lead a team of product managers
  • Deep expertise in AI training frameworks and LLMs
  • Strong communication and interpersonal skills
  • BS/MS in Computer Science or related field
Good to have:
  • Experience building LLM foundational models
  • Familiarity with LLM training frameworks (Megatron Core, NeMo)
  • Deep expertise with PyTorch/Inductor, JAX/XLA, OpenAI Triton
  • Understanding of state-of-the-art deep learning models
  • Knowledge of GPU architecture and performance profiling
Perks:
  • Equity
  • Benefits

Job Details

To bring AI to life, neural networks need to be trained at scale. Some of the popular AI models trained using NVIDIA platform are LLMs, Generative AI, Recommenders, Conversational virtual assistants. NVIDIA is at the forefront helping advance the latest research and product improvements for everybody! 

As a Director of Product - AI Training Platform Software, you will define product strategy and roadmap for NVIDIA's AI training softwares, including LLM pre-training and post-training frameworks. You will be partnering with the external framework developers such as PyTorch, JAX and related compilers and libraries to enable great user experience with optimized performance on GPUs. The Product Management organization is a strong, and sought after group both inside and outside of NVIDIA as our team is passionate about deep learning use cases to drive the company's strategy. We need a strong leader to continue to grow this area and work with customers, who are helping define the future of AI. We're looking for the rare blend of both technical and product skills and passion about groundbreaking technology. If this fits, we would love to learn more about you!

What You'll be Doing:

  • Develop product strategy and roadmap 

  • Work with the industry's leading AI researchers and developers to envision and develop the next generation of AI training framework

  • Partner with product marketing and campaign teams to drive go to market plan

  • Lead a strong team of product managers working in cross-functional areas to coordinate the overall product vision

What We Need to See:

  • BS or MS degree in Computer Science, Computer Engineering, or similar field or equivalent experience

  • 10+ overall nyears of technical product management, or similar, experience at a technology company

  • 5+ years of management experience

  • Deep expertise and work experience in AI training frameworks and LLMs

  • Strong communication and interpersonal skills

  • Ability to work physically at NVIDIA's Santa Clara headquarters and travel on occasion for partner meetings and events

Ways to Stand Out from the Crowd:

  • Experience building LLM foundational model and applying post-training techniques such as reinforcement learning.

  • Familiarity with LLM training framework such as Megatron Core, NeMo, PAXML, MaxText 

  • Deep expertise and work experience with PyTorch/Inductor, JAX/XLA and OpenAI Triton

  • Understanding of the state of the art deep learning model algorithms

  • Knowledge of GPU architecture, HW/SW co-design and performance profiling

The base salary range is 272,000 USD - 425,500 USD. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.

You will also be eligible for equity and benefits. NVIDIA accepts applications on an ongoing basis.

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.

Similar Jobs

Samsung Semiconductor - Intern, Machine Learning Engineer - VLMs

Samsung Semiconductor

San Jose, California, United States (Hybrid)
2 Months ago
ByteDance - Machine Learning Engineer Intern - Global Payment - 2025 Start

ByteDance

Singapore (On-Site)
1 Month ago
NXP - Junior Developer of Systems Testing Infrastructure

NXP

Brno, South Moravian Region, Czechia (On-Site)
6 Months ago
ByteDance - Student Researcher (Doubao (Seed) - Foundation Model - Video Generative Model)

ByteDance

San Jose, California, United States (On-Site)
19 Hours ago
Playrix - Feature Owner (LiveOps)

Playrix

Georgia (Remote)
5 Months ago
NVIDIA - Senior Developer Relations Manager - Robotics

NVIDIA

Tokyo, Japan (On-Site)
2 Months ago
NVIDIA - Solutions Architect, Generative AI

NVIDIA

Canada (On-Site)
1 Month ago
The Walt Disney Company - Lead Machine Learning Engineer

The Walt Disney Company

Bristol, Connecticut, United States (On-Site)
1 Day ago
NVIDIA - AI and ML Infra Software Engineer, GPU Clusters

NVIDIA

Santa Clara, California, United States (On-Site)
2 Months ago
NVIDIA - Software Engineer Intern - Mapping and Generative AI

NVIDIA

Shanghai, Shanghai, China (On-Site)
2 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Canva - Senior Computer Vision Engineer - Photo AI

Canva

Vienna, Vienna, Austria (Remote)
1 Week ago
Playrix - Game Director

Playrix

Armenia (Remote)
5 Months ago
NVIDIA - Deep Learning Solution Architect

NVIDIA

Shanghai, Shanghai, China (On-Site)
2 Months ago
NVIDIA - Solution Architect - CSP Cloud

NVIDIA

Beijing, Beijing, China (On-Site)
2 Months ago
ByteDance - Student Researcher (Doubao (Seed) - Foundation Model - Video Generative Model)

ByteDance

San Jose, California, United States (On-Site)
19 Hours ago
G5 Games - 2D UI/UX Artist (Hidden objects project)

G5 Games

Astana, Astana, Kazakhstan (Remote)
2 Months ago
Netflix - Machine Learning Engineer

Netflix

United States (Remote)
2 Months ago
ByteDance - Student Researcher (Doubao (Seed) - Foundation Model - Video Generative Model)

ByteDance

San Jose, California, United States (On-Site)
1 Day ago
Canva - Senior Computer Vision Engineer - Photo AI

Canva

Vienna, Vienna, Austria (Remote)
1 Month ago

Get notifed when new similar jobs are uploaded

Jobs in Santa Clara, California, United States

Next Level Business Services - Java Developer (Full Time)

Next Level Business Services

Littleton, Colorado, United States (On-Site)
5 Months ago
The Walt Disney Company - Manager, Product Design (Disney +)

The Walt Disney Company

Seattle, Washington, United States (On-Site)
4 Months ago
Next Level Business Services - Software Developer (Full Time)

Next Level Business Services

Redmond, Washington, United States (On-Site)
5 Months ago
Onward Search - Art Director

Onward Search

Westwood, Massachusetts, United States (On-Site)
22 Hours ago
Axinous - Account Executive, Enterprise - Bay Area

Axinous

California, United States (Remote)
3 Weeks ago
Thatgamecompany - Live Ops Engineer

Thatgamecompany

United States (Remote)
12 Hours ago
ByteDance - Senior/Tech Lead Software Development Engineer, Network Monitoring & Alerts - Seattle

ByteDance

Seattle, Washington, United States (On-Site)
4 Months ago
The Walt Disney Company - Associate Brand Art Director

The Walt Disney Company

San Francisco, California, United States (On-Site)
1 Day ago
Meta - Software Engineer, Systems ML - SW/HW Co-design

Meta

Sunnyvale, California, United States (Remote)
4 Months ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Dolby Laboratories - AIOps Research Scientist

Dolby Laboratories

Bengaluru, Karnataka, India (Hybrid)
5 Months ago
Razer - Solutions Architect

Razer

Singapore (On-Site)
5 Months ago
Egnyte - Principal Machine Learning Engineer - AI

Egnyte

India (Remote)
6 Days ago
Canva - Head of AI Research

Canva

San Francisco, California, United States (Remote)
2 Months ago
NVIDIA - Senior AI-HPC Storage Engineer

NVIDIA

Santa Clara, California, United States (On-Site)
2 Months ago
NVIDIA - Solutions Architect, Generative AI

NVIDIA

Canada (On-Site)
1 Month ago
Google - Senior Software Engineer, Machine Learning, Google Ads

Google

Mountain View, California, United States (On-Site)
4 Months ago
Lucid Reality Labs - ML/AI Engineer

Lucid Reality Labs

Poland (Remote)
11 Hours ago
Axon - Senior Technical Program Manager, AI

Axon

Seattle, Washington, United States (Remote)
1 Month ago

Get notifed when new similar jobs are uploaded

About The Company

Since its founding in 1993, NVIDIA (NASDAQ: NVDA) has been a pioneer in accelerated computing. The company’s invention of the GPU in 1999 sparked the growth of the PC gaming market, redefined computer graphics, ignited the era of modern AI and is fueling the creation of the metaverse. NVIDIA is now a full-stack computing company with data-center-scale offerings that are reshaping industry.


Hsinchu, Hsinchu City, Taiwan (On-Site)

Yokne'am Illit, North District, Israel (On-Site)

Seoul, South Korea (Hybrid)

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)

Ra'anana, Center District, Israel (On-Site)

Shanghai, Shanghai, China (On-Site)

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)

Be'er Sheva, South District, Israel (On-Site)

Santa Clara, California, United States (On-Site)

Santa Clara, California, United States (On-Site)

View All Jobs

Get notified when new jobs are added by NVIDIA

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug