Relational Foundation Model Engineer, Modern Data Stack

We are looking for a Relational Foundation Model Engineer to join our team at NVIDIA. You will collaborate with researchers and engineers to enhance our Transformer and GNN-based models to operate seamlessly over any relational schema and heterogeneous graph.

Responsibilities:

• Collaborate with researchers/engineers to enhance our Transformer and GNN-based models.
• Gain hands-on experience with high-impact use cases such as forecasting, entity matching, customer retention, and fraud detection.
• Leverage your knowledge in ML and AI to tackle real challenges.
• Contribute to scalable and adaptable solutions that push the boundaries of what's possible.
• Work may span the full lifecycle of modern ML systems.

Requirements:

• MS or PhD in Machine Learning, Computer Science, or equivalent program.
• 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.

Nice to Have:

• Familiarity with graph-based machine learning.
• Publications at venues such as NeurIPS, ICLR, ICML, or similar.