NVIDIA is seeking an engineer to join the Relational Foundation Model team, which focuses on developing a unified foundation model for understanding relational databases and heterogeneous graphs. The role involves designing, building, and evaluating novel Transformer and graph neural network architectures. This position offers the opportunity to work on impactful applications such as recommendation systems and fraud detection. Candidates will collaborate with top researchers and engineers throughout the machine learning lifecycle.

• MS or PhD in Machine Learning, Computer Science, or equivalent experience
• Proficiency in Python and deep learning frameworks, such as PyTorch
• At least 8 years of research experience in designing ML algorithm solutions
• Practical experience in using Predictive Models in Real World Applications

• Collaborate with researchers and engineers to enhance Transformer and GNN-based models for relational schemas and heterogeneous graphs.
• Gain hands-on experience with high-impact use cases such as forecasting and fraud detection.
• Leverage knowledge in ML and AI to tackle real challenges and contribute to scalable solutions.
• Contribute to the full lifecycle of modern ML systems, from architecture design to inference acceleration.

• Employees at NVIDIA are often offered comprehensive, day-one benefits—including medical, dental, and vision coverage with HSA support, life and disability insurance, an Employee Assistance Program, and a 401(k) with auto-enrollment. Many roles also have generous time off and holidays, donation matching (up to $10,000), and a wide menu of extras like FSAs, commuter benefits, legal and identity-theft protection, pet insurance, and wellness discounts. Optional programs can include student-loan and home-purchase support, plus family care resources and expert medical services.