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 advancing Physical AI, improving model performance and fidelity for applications 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 seamless user experiences. The ideal candidate will possess extensive experience in deep learning, model engineering, and cloud AI deployment, with a strong background in Python, PyTorch, and Hugging Face.
Must have:
  • 15+ years in deep learning/AI model development
  • Proficiency in model engineering (data curation, fine-tuning)
  • Strong software design & debugging skills
  • Advanced Python & PyTorch expertise
  • Deep understanding of AI system architectures
  • Cloud AI experience (AWS, Azure, GCP)
Good to have:
  • Industry impact in robotics or autonomous systems
  • Open-source contributions to AI frameworks
  • Published research in generative AI
  • Cross-disciplinary team leadership
  • GPU acceleration and performance optimization expertise
Perks:
  • Competitive salary
  • Generous benefits package

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

ByteDance - Machine Learning Engineer Graduate (AML Algorithm) - 2025 Start (PhD)

ByteDance

San Jose, California, United States (On-Site)
6 Months ago
Kaedim - Machine Learning Engineer

Kaedim

Singapore (On-Site)
10 Months ago
Inworld AI - Staff C++ Engineer

Inworld AI

Mountain View, California, United States (On-Site)
2 Months ago
Anavation - Senior Software Engineer

Anavation

Colorado Springs, Colorado, United States (Remote)
1 Month ago
Google - Senior Software Engineer, TPU, Google Cloud Platform

Google

Taipei City, Taiwan (On-Site)
1 Month ago
Inworld AI - AI Trainer (Contractor) - Writing & Gaming

Inworld AI

Vancouver, British Columbia, Canada (Remote)
2 Months ago
ByteDance - Senior Research Scientist, Foundation Model, Speech Understanding

ByteDance

San Jose, California, United States (On-Site)
6 Months ago
Google - Greenfield Artificial Intelligence Sales Specialist III

Google

San Francisco, California, United States (On-Site)
1 Month ago
Virtuos - Senior Games Tool Engineer (Machine Learning Specialist)

Virtuos

Shanghai, Shanghai, China (On-Site)
2 Months ago
NVIDIA - Director of Product - AI Training Platform Software

NVIDIA

Santa Clara, California, United States (On-Site)
2 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Amazon - Software Dev Engineer I, Amazon University Talent Acquisition

Amazon

Bengaluru, Karnataka, India (On-Site)
1 Month ago
Sleeper - Senior Frontend Engineer (Mobile)

Sleeper

Las Vegas, Nevada, United States (On-Site)
2 Months ago
Playrix - Senior Data Analyst (Marketing)

Playrix

Cyprus (Remote)
1 Month ago
CharacterAI - Research Engineer, ML Systems

CharacterAI

New York, New York, United States (On-Site)
2 Months ago
ByteDance - Senior Machine Learning Engineer - AML Algorithm

ByteDance

San Jose, California, United States (On-Site)
6 Months ago
Google - Software Engineer III, Security/Privacy, Google Cloud Compute Infrastructure

Google

Sunnyvale, California, United States (On-Site)
1 Month ago
Meta - Postdoctoral Researcher, Embodied AI (PhD)

Meta

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

Google

Sunnyvale, California, United States (On-Site)
6 Months ago
Epic Games - UI Engineering Director

Epic Games

(On-Site)
2 Months ago
Amanotes - Unity Developer (LiveOps Team)

Amanotes

Ho Chi Minh City, Ho Chi Minh City, Vietnam (On-Site)
4 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Santa Clara, California, United States

Google - Technical Delivery Infrastructure Engineer, Public Sector

Google

Reston, Virginia, United States (On-Site)
1 Month ago
NVIDIA - Speed Modeling and Prototyping Engineer

NVIDIA

Santa Clara, California, United States (Hybrid)
4 Months ago
Netflix - Senior Technical Writer

Netflix

Seattle, Washington, United States (On-Site)
1 Month ago
Snail Games - Staff Accountant (Onsite)

Snail Games

Loveland, Colorado, United States (On-Site)
6 Months ago
ByteDance - Software Engineer

ByteDance

San Jose, California, United States (On-Site)
1 Month ago
Netflix - Software Engineer L4, GenAI Platform

Netflix

United States (Remote)
3 Months ago
ByteDance - Engineering Manager Machine Learning Infrastructure

ByteDance

San Jose, California, United States (On-Site)
6 Months ago
ByteDance - Backend Software Engineer, Infrastructure Platform

ByteDance

New York, New York, United States (On-Site)
1 Month ago
Pattern® - Senior Ecommerce Manager

Pattern®

Lehi, Utah, United States (Hybrid)
8 Months ago
Google - Software Engineer, PhD

Google

Kirkland, Washington, United States (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Meta - Research Scientist Intern, Language and Multimodal Research for MetaAI (PhD)

Meta

Bellevue, Washington, United States (On-Site)
6 Months ago
Meta - Software Engineer, Machine Learning

Meta

Fremont, California, United States (Remote)
6 Months ago
Virtuos - Senior Machine Learning Engineer (Game)

Virtuos

Malaysia (On-Site)
1 Month ago
ByteDance - Software Engineer Intern (Doubao (Seed) - Machine Learning System) - 2025 Summer (MS)

ByteDance

Seattle, Washington, United States (On-Site)
6 Months ago
N-iX - Senior DS/AI Engineer

N-iX

Poland (Hybrid)
1 Month ago
Seedify - AI Product Manager

Seedify

(Remote)
4 Months ago
Google - Principal Technology Strategist, Cloud Space

Google

Sunnyvale, California, United States (On-Site)
1 Month ago
Passive Logic - AI Control Theory & Optimization Scientist

Passive Logic

Salt Lake City, Utah, United States (On-Site)
5 Months ago
NVIDIA - Software Engineering Intern, AI Engineering - 2025

NVIDIA

Shanghai, Shanghai, China (On-Site)
4 Months ago
Appier - Software Engineer, Machine Learning Platform

Appier

Taipei City, Taiwan (On-Site)
6 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