Principal DGX Cloud Machine Learning Architect Full-time

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.