Remote Principal Machine Learning Engineer - Large Scale Embedding - (Remote - US)

Description:

• The position is for a Principal Machine Learning Engineer specializing in Large Scale Embedding, working remotely from the United States.
• You will lead the development of large-scale, multi-entity embeddings that drive future recommendation systems.
• The role focuses on implementing cutting-edge architectures, such as Graph Neural Networks (GNN) and transformers, to model complex relationships within data.
• You will collaborate with cross-functional teams to design and scale machine learning pipelines, enabling efficient distributed training and serving.
• Your leadership will shape the technical roadmap and influence the future of machine learning applications at scale.
• Key accountabilities include leading the design and architecture of multi-entity embedding generation, defining the technical roadmap, developing and optimizing large-scale graph-based machine learning pipelines, ensuring scalable architectures for processing complex data, collaborating to improve relevance metrics, and mentoring your team.

Requirements:

• You must have 15+ years of technical leadership experience in machine learning and AI.
• A proven ability to lead ML initiatives and communicate complex ideas effectively to cross-functional teams is required.
• Expertise in Graph Neural Networks, collaborative filtering, knowledge graphs, and recommendation systems is essential.
• Strong coding proficiency in Python is necessary, along with experience in ML frameworks like PyTorch Geometric, DGL, TensorFlow, and scikit-learn.
• A solid understanding of ML infrastructure components and libraries for efficient distributed training and inference is required.

Benefits:

• The position offers comprehensive healthcare benefits, including medical, dental, and vision coverage.
• A 401(k) match is provided to help secure your financial future.
• Family planning support, including gender-affirming care, is available.
• Mental health and coaching benefits are offered to support well-being.
• Flexible vacation and global days off are provided to maintain work-life balance.
• Generous paid parental leave and paid volunteer time off are included as part of the benefits package.