Principal Machine Learning Engineer - Large Scale Embedding

Overview

Role involves leading the design and architecture of multi-entity embedding generation using GNN and transformers.

Ideal candidate should have 15+ years of experience in technical leadership with expertise in machine learning and graph neural networks.

Locations

• United States

Requirements

• 15+ years of technical leadership experience
• Expertise in Graph Neural Networks
• Strong coding skills in Python
• Experience with ML frameworks like PyTorch and TensorFlow

Responsibilities

• Lead team in architecting GNN and transformers
• Define technical roadmap
• Develop large-scale ML pipelines
• Collaborate with cross-functional teams
• Enable distributed GPU training
• Lead feature engineering efforts
• Contribute to team strategy

Benefits

• Comprehensive healthcare benefits
• 401k match
• Family planning support
• Mental health & coaching benefits
• Flexible vacation
• Generous paid parental leave
• Paid volunteer time off