[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

• NVIDIA is 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.