Principal DGX Cloud Machine Learning Architect

3 Months ago • 15 Years + • Artificial Intelligence • $272,000 PA - $425,500 PA

Job Summary

Job Description

NVIDIA seeks a Principal DGX Cloud Machine Learning Architect to optimize generative AI models (VLMs, WFMs, CV models) for DGX Cloud customers. This role focuses on enhancing Physical AI applications, improving model performance and fidelity for tasks like classification, auto-labeling, and autonomous vehicle policies. Responsibilities include developing and optimizing generative AI models using NVIDIA's AI software stack, designing evaluation methodologies, collaborating with cross-functional teams, analyzing and improving model performance, and architecting scalable software platforms for DGX Cloud customers. The ideal candidate will have extensive experience in deep learning, model engineering, and cloud AI deployment, with a strong understanding of PyTorch, Hugging Face, and AI system architectures.
Must have:
  • 15+ years deep learning/AI model development experience
  • Proficiency in model engineering (data curation, fine-tuning)
  • Strong software design & debugging skills, performance analysis
  • Advanced Python & PyTorch, experience with ML tools (Hugging Face)
  • Deep understanding of algorithms, programming, AI architectures
  • Cloud AI expertise (AWS, Azure, GCP), performance optimization
Good to have:
  • Industry impact in robotics/autonomous systems
  • Open-source contributions to AI frameworks
  • Published research in generative AI/Physical AI
  • Cross-disciplinary team leadership experience
  • GPU acceleration, hardware/software co-design expertise

Job Details

NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years. It’s a unique legacy of innovation that’s fueled by great technology—and amazing people. Today, we’re tapping into the unlimited potential of AI to define the next era of computing. An era in which our GPU acts as the brains of computers, robots, and self-driving cars that can understand the world. Doing what’s never been done before takes vision, innovation, and the world’s best talent. As an NVIDIAN, you’ll be immersed in a diverse, supportive environment where everyone is inspired to do their best work. Come join the team and see how you can make a lasting impact on the world.

NVIDIA DGX™ Cloud is a scalable, end-to-end AI platform, co-engineered with leading cloud service providers (CSPs) to deliver groundbreaking performance for developers. We are seeking a Principal DGX Cloud Machine Learning Architect to optimize generative AI models, including Vision Language Models (VLMs), World Foundation Models (WFMs), and Computer Vision (CV) models, for DGX Cloud customers. This role is crucial for advancing Physical AI, the next frontier in AI. Your expertise will improve model performance and fidelity for applications like classification, auto-labeling, and robotic/autonomous vehicle policies. You will help DGX Cloud customers achieve outstanding results at scale. You will work across the entire machine learning stack, from high-level frameworks like PyTorch and Hugging Face to model evaluation, data curation, fine-tuning, and re-architecture. This is an outstanding opportunity for an innovative ML engineer at the intersection of research and engineering to drive impact in a high-performance AI environment.

What You’ll Be Doing:

  • Develop and optimize innovative generative AI models such as VLMs and WFMs using NVIDIA’s AI software stack for Physical AI applications.

  • Design and implement sophisticated evaluation methodologies and automation techniques to streamline model assessments.

  • Collaborate with world-class teams across NVIDIA to refine and improve foundation models for AI-powered solutions.

  • Analyze, profile, and improve model performance to drive efficiency, scalability, and precision.

  • Architect scalable, modular software platforms that improve AI adoption by DGX Cloud customers, ensuring seamless user experiences and broad model support.

What We Need to See:

  • Master’s, PhD, or equivalent experience in Computer Science, AI, Applied Math, or a related field (or equivalent experience).

  • 15+ years of demonstrated ability in deep learning, AI model development, or research.

  • Proficiency in model engineering, including data curation, fine-tuning, and evaluation.

  • Strong software design and debugging skills, with expertise in performance analysis and test design.

  • Advanced Python and PyTorch proficiency, with experience in ML tools such as Hugging Face.

  • Deep understanding of algorithms, programming fundamentals, and AI system architectures.

  • Excellent written and verbal communication skills, with the ability to work independently and collaboratively in a multifaceted environment.

  • Cloud AI Expertise: Hands-on experience optimizing and deploying AI systems at scale on major cloud platforms (AWS, Azure, GCP), focusing on performance and cost-efficiency.

  • Automation & Scalability: Proven track record to build automated evaluation methodologies and scalable data curation pipelines to continuously boost model performance.

Ways to Stand Out from the Crowd:

  • Industry Impact: A consistent record of transitioning groundbreaking AI research into production, particularly in robotics or autonomous systems.

  • Open-Source Leadership: Active contributions to key AI frameworks (e.g., PyTorch, Hugging Face) or other significant open-source projects.

  • Thought Leadership: Published research, patents, or recognition in generative AI, Physical AI, or related fields that highlight your ability to drive innovation.

  • Cross-Disciplinary Leadership: Experience leading multi-functional teams (engineering, research, product) to deliver coordinated AI solutions.

  • Performance Optimization Mastery: Expertise in GPU acceleration, hardware/software co-design, and deep performance optimization for AI workloads.

With competitive salaries and a generous benefits package (www.nvidiabenefits.com ), we are widely considered to be one of the technology world’s most desirable employers. We have some of the most forward-thinking and hardworking people in the world working for us and, due to outstanding growth, our best-in-class engineering teams are rapidly growing. If you're a creative and autonomous engineer with a real passion for technology, we want to hear from you!

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

NVIDIA - Senior Developer Technology Engineer, Public Sector

NVIDIA

Washington, District Of Columbia, United States (Remote)
2 Months ago
ByteDance - Research Engineer (Machine Learning Training System) - 2025 Start

ByteDance

Singapore (On-Site)
7 Months ago
Sleeper - Senior Frontend Engineer (Mobile)

Sleeper

Las Vegas, Nevada, United States (On-Site)
1 Month ago
ByteDance - Research Scientist- Foundation Model, Generative AI

ByteDance

Seattle, Washington, United States (On-Site)
6 Months ago
Google - PhD Software Engineer

Google

Sunnyvale, California, United States (On-Site)
1 Month ago
Level AI - Backend Engineer - Customer Engineering

Level AI

Noida, Uttar Pradesh, India (Hybrid)
7 Months ago
DNEG - Head of Machine Learning

DNEG

London, England, United Kingdom (Remote)
1 Month ago
Google - Customer Engineer, Machine Learning, Google Cloud

Google

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Microsoft - Research Intern - Microsoft Teams CMD Labs

Microsoft

Redmond, Washington, United States (On-Site)
1 Month ago
Google - Software Engineer III, Security Endpoint Agents, Core

Google

Dublin, County Dublin, Ireland (On-Site)
1 Month ago
Inworld AI - Senior C++ Developer - Canada

Inworld AI

Vancouver, British Columbia, Canada (On-Site)
7 Months ago
Twitch - Software Engineer I - iOS

Twitch

New York, New York, United States (On-Site)
4 Months ago
Epic Games - Senior Rendering Engineer

Epic Games

London, England, United Kingdom (On-Site)
1 Month ago
Tencent - Research Intern (NLP)

Tencent

Palo Alto, California, United States (On-Site)
3 Months ago
Google - Software Engineer III, Infrastructure, Google Cloud Storage

Google

Kirkland, Washington, United States (On-Site)
1 Month ago
PlayStation Global - Staff Machine Learning Engineer, Anomaly Detection

PlayStation Global

London, England, United Kingdom (Hybrid)
6 Months ago
Google - Senior Software Engineer, Storage, Pixel Software

Google

New Taipei, New Taipei City, Taiwan (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Jobs in Canada

PwC - Forensic Manager

PwC

Toronto, Ontario, Canada (On-Site)
7 Months ago
Activision - Senior Network Engineer

Activision

Vancouver, British Columbia, Canada (On-Site)
2 Months ago
NetEase Games - Global Influencer Marketing

NetEase Games

Canada (On-Site)
5 Months ago
Evolution - Customer Service - Japanese Speaking Game Presenter (No experience required)

Evolution

Burnaby, British Columbia, Canada (On-Site)
8 Months ago
Granicus - SLED Enterprise Account Executive - Western Canada (Local)

Granicus

Canada (Remote)
7 Months ago
Rockstar Games - Animation Systems Programmer

Rockstar Games

Oakville, Ontario, Canada (On-Site)
1 Month ago
Epic Games - Character Concept Outsource Lead (IP)

Epic Games

Montreal, Quebec, Canada (On-Site)
1 Month ago
Rockstar Games - Gameplay Animator (Mid/Senior)

Rockstar Games

Oakville, Ontario, Canada (On-Site)
2 Months ago
Ubisoft - Senior Analyst, Live Analytics

Ubisoft

Montreal, Quebec, Canada (On-Site)
1 Month ago
Epic Games - Senior AI Gameplay Programmer, UEFN

Epic Games

Montreal, Quebec, Canada (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Canva - Senior Machine Learning Engineer - Photo AI

Canva

Prague, Czechia (Remote)
4 Months ago
Trend Micro - NLP / Prompt Engineer (VicOne_Automotive Security)

Trend Micro

Taipei City, Taiwan (On-Site)
7 Months ago
Google - Senior Research Engineer, AI/ML

Google

London, England, United Kingdom (On-Site)
1 Month ago
Google - Staff Software Engineer, GPU Performance, Core ML

Google

London, England, United Kingdom (On-Site)
1 Month ago
Ubisoft - Senior C++ Programmer - Machine Learning

Ubisoft

Montreal, Quebec, Canada (On-Site)
2 Months ago
Evolution - Data Scientist

Evolution

Warsaw, Masovian Voivodeship, Poland (Hybrid)
9 Months ago
Google - Customer Engineer, Data and AI

Google

Melbourne, Victoria, Australia (On-Site)
1 Month ago
Hyper Verge - Machine Learning Engineer II

Hyper Verge

Bengaluru, Karnataka, India (On-Site)
7 Months ago
Google - Senior ML Compiler Engineer, Silicon

Google

Bengaluru, Karnataka, India (On-Site)
1 Month ago
NVIDIA - Senior Software Engineer, AI Resiliency

NVIDIA

Redmond, Washington, United States (On-Site)
2 Months 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.

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

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

Massachusetts, United States (On-Site)

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

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

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

Texas, United States (On-Site)

Santa Clara, California, United States (Hybrid)

Santa Clara, California, United States (Hybrid)

Pune, Maharashtra, India (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