Principal DGX Cloud Machine Learning Architect

Job Description

NVIDIA is seeking a Principal DGX Cloud Machine Learning Architect to optimize generative AI models for Physical AI applications.

Requirements

• Master’s, PhD, or equivalent experience in Computer Science, AI, Applied Math, or a related field
• 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
• Hands-on experience optimizing and deploying AI systems at scale on major cloud platforms, focusing on performance and cost-efficiency
• Proven track record to build automated evaluation methodologies and scalable data curation pipelines to continuously boost model performance

Responsibilities

• Develop and optimize generative AI models for Physical AI applications using NVIDIA’s AI software stack
• Design and implement evaluation methodologies and automation techniques to streamline model assessments
• Collaborate with 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 to improve AI adoption by DGX Cloud customers

Benefits

• Multiple relocation packages
• Two weeklong shutdowns (mid-summer and year-end) in the US (in addition to PTO)
• 8-week parental leave
• 9 Employee Resource Groups
• Annual bonus offering
• Flexible work arrangements
• Up to 6% 401K matching