Relational Foundation Model Engineer, Modern Data Stack

NVIDIA
Santa Clara
Workplace: OnsiteFull timeUSD 184,000 - 356,500 annuallyFunction: Healthcare (Clinical, Medical, Wellness)Experience: 8+ yearsEducation: mastersSkills: ["Collaboration","Creativity","Autonomy","Problem-solving"]

Build and evaluate a unified relational foundation model that generalizes across relational schemas and heterogeneous graphs. Design and implement novel Transformer and graph neural network architectures for tasks such as forecasting, entity matching, retention, fraud detection, and predictive maintenance. Partner across the full ML lifecycle—from architecture exploration and large-scale training to post-training optimization and inference acceleration—optimized for NVIDIA-accelerated infrastructure.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
NVIDIA
NVIDIA
1 day ago

Relational Foundation Model Engineer, Modern Data Stack

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 26 minutes agoStatus: Live

Job Summary

Build and evaluate a unified relational foundation model that generalizes across relational schemas and heterogeneous graphs. Design and implement novel Transformer and graph neural network architectures for tasks such as forecasting, entity matching, retention, fraud detection, and predictive maintenance. Partner across the full ML lifecycle—from architecture exploration and large-scale training to post-training optimization and inference acceleration—optimized for NVIDIA-accelerated infrastructure.
Location: Santa Clara
Workplace: Onsite
Employment Type: Full time
Job Function: Healthcare (Clinical, Medical, Wellness)

Key Responsibilities

  • •Enhance Transformer and GNN-based models to operate across any relational schema and heterogeneous graph.
  • •Collaborate with researchers and engineers across the ML lifecycle, from architecture exploration to large-scale training and post-training optimization.
  • •Build scalable and adaptable solutions that generalize beyond single-table benchmarks for relational reasoning and graph learning.
  • •Develop and apply the foundation model to high-impact use cases such as forecasting, entity matching, customer retention, and fraud detection.
  • •Contribute to next-generation relational foundation model development, including inference acceleration where needed.

Pay and Benefits

Salary: USD 184,000 - 356,500 annually
Equity and Bonus:Equity

Key Requirements

  • •MS or PhD in Machine Learning or Computer Science, or equivalent experience.
  • •Proficiency in Python and deep learning frameworks such as PyTorch.
  • •At least 8 years of research experience designing ML algorithm solutions.
  • •Practical experience using predictive models in real-world applications.
  • •Familiarity with graph-based machine learning and publications at venues such as NeurIPS, ICLR, ICML, or similar (to stand out).
Experience:8+ yearsResearchMachine learning
Education:Master's in Machine Learning, Computer Science
Skills:CollaborationCreativityAutonomyProblem-solving
Tech Stack:PythonPyTorchTransformerGraph neural networksGNNDeep learningNeurIPSICLRICML

Company Brief

NVIDIA
Designs and manufactures GPUs, AI accelerators, and system-on-chip products for gaming, data centers, professional visualization, and automotive markets, enabling advanced graphics, AI, and high-performance computing solutions worldwide.
Industry: Electronics Manufacturing
Company Size: Enterprise (1,001+ employees)
Revenue: USD 1B+
Growth: Public Company
Valuation: Public Company (Market Cap in USD)
Funding: IPO / Publicly Listed
Headquarters: Santa Clara, United States
Founded: 1993
Glassdoor
Glassdoor: 4.3
WebsiteLinkedInGlassdoor