Relational Foundation Model Engineer, Modern Data Stack
$184k - $357k • US, CA, Santa Clara • Full-time
Posted 1d ago
Remote Work Policy
On-site
Employment Type
Full-time
Categories
LLM Engineer
About the job
NVIDIA is building a unified foundation model that understands relational databases and heterogeneous graphs, enabling a new era of enterprise AI. This role involves designing, building, and evaluating novel Transformer and graph neural network architectures that generalize across diverse data schemas. The work will power applications like recommendation systems, demand forecasting, fraud detection, and predictive maintenance. You will collaborate with researchers and engineers across the entire machine learning lifecycle, from architecture exploration and large-scale training to post-training optimization and high-performance inference, translating foundational research into production systems.
Responsibilities
- Collaborate with researchers and engineers to enhance Transformer and GNN-based models for relational schemas and heterogeneous graphs.
- Gain hands-on experience with high-impact use cases like forecasting, entity matching, customer retention, and fraud detection.
- Leverage ML and AI knowledge to solve real-world challenges with scalable and adaptable solutions.
- Contribute to the full lifecycle of ML systems, including architecture design, training, post-training optimization, and inference acceleration.
- Contribute to the next generation of the Relational Foundation Model.
Requirements
- MS or PhD in Machine Learning, Computer Science, or equivalent experience.
- Proficiency in Python and deep learning frameworks like PyTorch.
- At least 8 years of research experience in designing ML algorithm solutions.
- Practical experience using predictive models in real-world applications.
Benefits
- Equity
- Benefits