Improve Reddit Relevance by developing and implementing large-scale graph-based machine learning pipelines for recommendation systems.
Requirements
- Expertise in Graph Neural Networks, collaborative filtering, knowledge graphs, and deep learning for recommendations.
- Understanding of graph theory, network science, and representation learning technique
- Strong coding skills in Python and experience with ML frameworks like PyTorch Geometric (PyG), Deep Graph Library (DGL), TensorFlow, and scikit-learn.
- Solid understanding of ML infrastructure components and libraries (data parallel, model parallel, pipeline parallel, torch.inductor, model pruning, etc.) enabling efficient distributed training and inference.
- Proven ability to lead ML initiatives, mentor engineers, and communicate complex concepts to cross-functional teams.
- 15+ years of Technical Leadership Experience
- Experience with distributed GPU based training and online serving architecture
Responsibilities
- Lead the team that architects and designs GNN and transformers based multi-entity embedding generation.
- Develop and optimize large-scale graph-based machine learning pipelines for recommendation systems.
- Architect scalable and efficient GNN and transformers-based recommendation models that can process complex, interconnected data structures.
- 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 to be used for creating embeddings
- Contribute towards team and product strategy, operations and execution at Reddit.
- Collaborate with cross functional business units such as Ads teams leveraging the models for upstream functions and improve relevance metrics.
Other
- 15+ years of Technical Leadership Experience
- Proven ability to lead ML initiatives, mentor engineers, and communicate complex concepts to cross-functional teams.
- Comprehensive Healthcare Benefits and Income Replacement Programs
- 401k Match
- Family Planning Support
- Gender-Affirming Care
- Mental Health & Coaching Benefits
- Flexible Vacation & Reddit Global Days off
- Generous paid Parental Leave
- Paid Volunteer time off
