Machine Learning Engineer

NVIDIA is seeking a Machine Learning Engineer to join its team. You will collaborate with researchers and engineers to enhance Transformer and GNN-based models, gain hands-on experience with high-impact use cases, and apply your knowledge in ML and AI to address real challenges.

Responsibilities:

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

Requirements:

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

Nice to Have:

• Familiarity with graph-based machine learning; publications at venues such as NeurIPS, ICLR, ICML, or similar.