TL;DR

Principal Machine Learning Engineer (AI): Leading the design and architecture of GNN and transformers-based multi-entity embedding generation with an accent on building scalable recommendation systems. Focus on developing large-scale graph-based machine learning pipelines and collaborating with cross-functional teams to enhance relevance metrics.

Location: Remote - United States

Salary: $276,700 - $387,400

Company

%hirify_global% is a community of communities, home to the most open and authentic conversations on the internet.

What you will do

Lead the team that architects and designs GNN and transformers based multi-entity embedding generation.
Define the technical roadmap and plan of execution in collaboration with cross-functional partners.
Develop and optimize large-scale graph-based machine learning pipelines for recommendation systems.
Collaborate with ML Infrastructure teams to enable distributed GPU based training and online serving architecture.
Lead feature engineering efforts to identify and curate expressive raw data for creating embeddings.
Mentor and advocate for the team.

Requirements

15+ years of Technical Leadership Experience.
Expertise in Graph Neural Networks, collaborative filtering, and deep learning for recommendations.
Strong coding skills in Python and experience with ML frameworks like PyTorch Geometric, TensorFlow, and scikit-learn.
Solid understanding of ML infrastructure components enabling efficient distributed training and inference.