[Hiring] Relational Foundation Model Intern @NVIDIA

Role Description

NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years. Today, we’re tapping into the unlimited potential of AI to define the next era of computing.

• Collaborate with researchers/engineers to enhance our Transformer and GNN-based models to operate seamlessly over any relational schema and heterogeneous graph.
• Gain hands-on experience with high-impact use cases such as forecasting, entity matching, customer retention, and fraud detection – all built on top of a single, extensible foundation model.
• Apply your knowledge in ML and AI to address real challenges while contributing to scalable and adaptable solutions that push the boundaries of what’s possible.
• Work may span the full lifecycle of modern ML systems: from architecture design/training to post-training optimization and inference acceleration.

Qualifications

• Pursuing MS or PhD in Machine Learning, Computer Science, or equivalent program area.
• Proficiency in Python and deep learning frameworks, such as PyTorch.
• Coursework or research experience in crafting ML algorithm solutions.

Requirements

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

Benefits

• Internship pay is based on the position, your location, year in school, degree, and experience.
• Pay is pro-rated for the duration of the internship.
• The expected annual rates for this position are:
• Germany: 47,850 EUR – 82,940 EUR.
• France: 40,500 EUR – 70,200 EUR.