← All roles

Relational Foundation Model Engineer, Modern Data Stack

NVIDIA· Engineer, Sys SW
Apply on company site
Santa Clara, CA$184k - $287.5kFull-timePosted 2mo ago

About the role

Engineer role focused on designing, building, and evaluating Transformer and graph neural network architectures for relational schemas and heterogeneous graphs, with production-ready ML systems and high-performance inference on NVIDIA infrastructure.

Responsibilities

  • Collaborate with researchers/engineers to enhance Transformer and GNN-based models to operate over relational schemas and heterogeneous graphs
  • Gain hands-on experience with use cases such as forecasting, entity matching, customer retention and fraud detection built on a single foundation model
  • Leverage ML/AI knowledge to tackle real-world challenges with scalable, adaptable solutions
  • Work across the full lifecycle of modern ML systems from architecture design/training to post-training optimization and inference acceleration
  • Contribute to the next generation of the Relational Foundation Model.

Qualifications

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

Skills

Required:Transformer architecturesGraph Neural Networks (GNNs)PythonPyTorch

Benefits

  • equity and benefits (as per company policy)
Last updated Aug 21, 2026. Interested? Apply on the company site.

About NVIDIA

World Leader in AI Computing

NVIDIA invents the GPU and drives advances in AI, HPC, gaming, creative design, autonomous vehicles, and robotics.

AIautonomous vehiclesgaminggpugraphics cardhigh performance computingliquid-ai-recruiting-targetllm
Company profileWebsite

More open roles at NVIDIA