Relational Foundation Model Intern — Graph AI & Transformers

NVIDIA; United States; Remote; USD 56.000 - 78.000; Vollzeit

Zusammenfassung

NVIDIA is seeking a highly motivated ML research intern in the United States to collaborate with researchers and engineers on transformer and GNN-based models. You will work on a scalable foundation model across relational schemas and heterogeneous graphs.

You should be pursuing an MS or PhD in ML, CS, or related field, with strong Python and PyTorch skills and experience in ML algorithm development. This internship offers exposure to cutting-edge AI research and real-world applications.

Qualifikationen

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

Aufgaben

• Collaborate with researchers/engineers to enhance 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.

Kenntnisse

Python

PyTorch

Machine Learning

Ausbildung

MS or PhD in ML/CS

Tools

PyTorch

Jobbeschreibung

NVIDIA is seeking a highly motivated ML research intern in the United States to collaborate with researchers and engineers on transformer and GNN-based models. You will work on a scalable foundation model across relational schemas and heterogeneous graphs.

You should be pursuing an MS or PhD in ML, CS, or related field, with strong Python and PyTorch skills and experience in ML algorithm development. This internship offers exposure to cutting-edge AI research and real-world applications.